Then you could simply run python3 train.py Experiments:įor each class, I take the first 10000 images, and then split them to training and test sets with ratio 8:2. If you want to train your model with different list of categories, you only need to change the constant CLASSES at src/config.py and download necessary npz files. You need to download npz files corresponding to 20 classes my model used and store them in folder data.

You could find my trained model at trained_models/whole_model_quickdraw Training The table below shows 20 categories my model used: Here I only picked up 20 files for 20 categories Categories: The dataset used for training my model could be found at. The script and demo will be released soon Dataset whether youre working on a school poster or brainstorming your next comic book character. When you want to stop drawing, press space againīelow is the demo by running the sript camera_app.py: sketchpad is available online and for download on pc and mac. When you are ready for drawing, you need to press space button. You draw, and a neural network tries to guess what you’re drawing.

When the pen (object) appears in front of camera, it will be catched and highlighted by an yellow circle. This is a game built with machine learning. In order to use this app, you need a pen (or any object) with blue, red or green color.

Run an app which you could draw on a canvas.

Run an app which you could draw in front of a camera (If you use laptop, your webcam will be used by default).

Here is my python source code for QuickDraw - an online game developed by google.

Quick draw online download