Skip to content

A Doodle Classifier written in python, using NumPy, pygame and google quick draw data set.

Notifications You must be signed in to change notification settings

aryahassibi/Doodle-Classifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 
 
 

Repository files navigation

Doodle Classifier

A Doodle Classifier written in python(3.7.6), using NumPy(1.18.1), pygame(1.9.6) and Google Quick Draw data set.

I used my own Multilayer Neural Netwrok to create this project, you can read about it on its page.

DoodleClassifierGIF

How To Use DoodleClassifier.py

After downloading the file create a folder named *data at the same place you are keeping DoodleClassifier.py. Then you have to download two or more numpy bitmap files from Google Quick Draw data set

note: you can search the name of the category that you are interested in the search bar at the top of the page.

For more information about the Google Quick Draw data set, check out this page.

Now that you have downloaded .npy files that you wanted, you have to put them in data folder.

note: names of the categories in the program are based on the name of the files, so it's better not to change the name of the files.

After doing all of this, you are ready to run the program and enjoy it.

How Does DoodleClassifier.py work?

When you run the program it starts loading the data set and then trains the neural network(this step may take a few seconds).
After training the neural network it shows you the canvas and you can draw whatever you like on the canvas, then the neural network makes a guess based on the thing that you have drawn and you can see its prediction in the red text box at the bottom of the screen.
If you want to clear the canvas you can press the clear button in the bottom left corner of the screen.

note: For further details read the comments in DoodleClassifier.py.

About

A Doodle Classifier written in python, using NumPy, pygame and google quick draw data set.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages