Skip to content
master
Switch branches/tags
Code

Latest commit

Akshay Bahadur and Akshay Bahadur Updated structure

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time

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

  1. Apple 🍎
  2. Bowtie πŸŽ€
  3. Candle πŸ•―οΈ
  4. Door πŸšͺ
  5. Envelope βœ‰οΈ
  6. Fish 🐟
  7. Guitar 🎸
  8. Ice Cream 🍦
  9. Lightning ⚑
  10. Moon πŸŒ›
  11. Mountain πŸ—»
  12. Star ⭐
  13. Tent β›Ί
  14. Toothbrush 🧹
  15. Wristwatch ⌚

Python Implementation πŸ‘¨β€πŸ”¬

  1. Network Used- Convolutional Neural Network

If you face any problem, kindly raise an issue

Setup πŸ–₯️

  1. Get the dataset as mentioned above and place the .npy files in /data folder.
  2. First, run LoadData.py which will load the data from the /data folder and store the features and labels in pickel files.
  3. 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.
  4. Now you need to have the data, run QuickDrawApp.py which will use use the webcam to get what you have drawn.
  5. For altering the model, check QD_trainer.py.
  6. 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.

Mergerd to Google's git repo 🌼

See the pull request here

Execution πŸ‰

python3 QuickDrawApp.py

Results πŸ“Š

References: πŸ”±

About

A simple implementation of Google's Quick, Draw Project for humans. πŸ–ŒοΈ πŸ–ΌοΈ

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages