Guess My Drawing
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.
Training the Model
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