https://github.com/hunterkalinoski/doodle-detector

A website which hosts a neural network trained to predict your drawings.
https://github.com/hunterkalinoski/doodle-detector

nextjs react tensorflow

Last synced: 3 months ago
JSON representation

A website which hosts a neural network trained to predict your drawings.

• Host: GitHub
• URL: https://github.com/hunterkalinoski/doodle-detector
• Owner: hunterkalinoski
• Created: 2022-11-08T03:35:35.000Z (over 3 years ago)
• Default Branch: main
• Last Pushed: 2023-03-06T23:19:23.000Z (over 3 years ago)
• Last Synced: 2025-03-11T13:52:36.930Z (over 1 year ago)
• Topics: nextjs, react, tensorflow
• Language: Jupyter Notebook
• Homepage: http://doodle-detector.vercel.app
• Size: 5.55 MB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

README

          
# Doodle Detector

A website that predicts your drawings!  

Inspired by Google's [Quick,Draw!](https://quickdraw.withgoogle.com/), and Sebastian Lague's [Neural Network video](https://www.youtube.com/watch?v=hfMk-kjRv4c&t)

There are two underlying neural networks.  These were created with TensorFlow and Keras (Python).  

The notebook for the MNIST NN can be found in the CNN - Augmented directory.  

The other notebook was not uploaded because it required a large local datasource, but is very similar.

One model was trained on the MNIST dataset, and can predict digits (0-9).  

The other model was trained on Google's Quick,Draw! dataset, and can predict the following shapes:  

- circle

- star

- triangle

- umbrella

- basketball

- fish

- house

- cat

Simply visit [the website](https://doodle-detector.vercel.app/), choose which model you want to use, and begin drawing!