Guess My Drawing classifies drawn images, currently supporting 345 different categories. It displays the top-5 confidence categories. I trained the model on Google's Quick, Draw! dataset, and use iOS 11's CoreML framework to run the trained model.
- Download or clone this repository.
- Open the project file in Xcode.
- Build and install the application to your device.
Here is a GIF of it classifying a few drawings correctly:
I am currently not releasing the details on how I trained the model. The architecture of the model is:
convolution (3x3 window, 64 filters) w/ relu
convolution (3x3 window, 32 filters) w/ relu
convolution (5x5 window, 32 filters) w/ relu
fully connected (1024 neurons) w/ relu
fully connected (512 neurons) w/ relu
fully connected (512 neurons) w/ relu
fully connected (345 neurons) w/ softmax