CNN classifier for hand drawn sketchs written with TensorFlow
You can prepare your own dataset with prepare_data.py. The structure should look like this:
./data/class1/1.png
./data/class2/1.png
...
Here is a good dataset.
cnn.py parameters:
-s: save model as chekpoint file
-r: restore checkpoint file
-f: run on floydhub