Skip to content

Hydrino/Doodly

Repository files navigation

Doodly

Can a neural network identify what you just doodled?

Draw any doodle from a hat, tree, an apple or an alarm clock. The neural network would identify which one of them did you draw.

Architecture

I have used Convolutional Neural Network(CNN) for classifying the doodle. CNN's work well for images instead of vanilla neural network(fully connected). This is because of CNN's ability to take into account the locality of pixels. I have created this architecture in keras and the code for this is available in the train.py file.

Frontend

I have encorporated this CNN into an Android app. I trained the CNN separately and then freezed the tensorflow model and imported it in Android for inference.

alt text

alt text