๐๏ธ ย ๐ผ๏ธ
Quick, Draw
Can a neural network learn to recognize doodling? Quick, Draw
๐ฆ
Code Requirements You can install Conda for python which resolves all the dependencies for machine learning.
pip install -r requirements.txt
๐จโ๐จ
Description Quick, Draw! is an online game developed by Google that challenges players to draw a picture of an object or idea and then uses a neural network artificial intelligence to guess what the drawings represent. The AI learns from each drawing, increasing its ability to guess correctly in the future.The game is similar to Pictionary in that the player only has a limited time to draw (20 seconds).The concepts that it guesses can be simple, like 'foot', or more complicated, like 'animal migration'. This game is one of many simple games created by Google that are AI based as part of a project known as 'A.I. Experiments'. Quick, Draw
๐๏ธ
Dataset Follow the documentation here to get the dataset. I got .npy
files from google cloud for 14 drawings.
Labels
- Apple
๐ - Bowtie
๐ - Candle
๐ฏ๏ธ - Door
๐ช - Envelope
โ๏ธ - Fish
๐ - Guitar
๐ธ - Ice Cream
๐ฆ - Lightning
โก - Moon
๐ - Mountain
๐ป - Star
โญ - Tent
โบ - Toothbrush
๐งน - Wristwatch
โ
๐จโ๐ฌ
Python Implementation - Network Used- Convolutional Neural Network
If you face any problem, kindly raise an issue
๐ฅ๏ธ
Setup - Get the dataset as mentioned above and place the
.npy
files in/data
folder. - First, run
LoadData.py
which will load the data from the/data
folder and store the features and labels in pickel files. - Now you need to have the data, run
QD_trainer.py
which will load data from pickle and augment it. After this, the training process begins. - Now you need to have the data, run
QuickDrawApp.py
which will use use the webcam to get what you have drawn. - For altering the model, check
QD_trainer.py
. - For tensorboard visualization, go to the specific log directory and run this command
tensorboard --logdir=.
You can go tolocalhost:6006
for visualizing your loss function and accuracy.
๐ผ
Mergerd to Google's git repo See the pull request here
๐
Execution python3 QuickDrawApp.py