You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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 to localhost:6006 for visualizing your loss function and accuracy.