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