Watch a neural network classify your fruit drawings. GUI was made using Electron.js paired up with a Python backend running a flask server.

Watch the video here : https://youtu.be/O8Gvkq8y-qs

Google colab notebook : https://colab.research.google.com/drive/1pG7gbXyAq-8UL_Pj18FV6O-jU0MmYZze

• Python
• Electron.js
• node.js
• Tensorflow
• Keras
• numpy
• scikit-learn
• PIL
• scipy
• Flask

Download the dataset inside a 'data/' directory. Dataset : https://github.com/googlecreativelab/quickdraw-dataset (get the numpy bitmaps)

$ python train.py

After training, link your saved model in server.py and then run the server

$ python server.py

After the server is up and running,

$ npm start