This project implements doodle recognition using a MobileNet model trained on the Google QuickDraw dataset. It allows users to perform recognition on imported images.
Data found here
- Train a MobileNet model for doodle recognition using the QuickDraw dataset.
- Recognize human-drawn doodles from
.png
and.jpg
images
- Python 3.x
- TensorFlow
- NumPy
- OpenCV
- Pillow (optional, used for image processing)
- Matplotlib (optional, used for data visualization)
-
Clone the repository:
git clone https://github.com/AadiJo/Doodle-Recognition.git
-
Navigate to the project directory:
cd 'your-repo'
-
Install the dependencies:
pip install -r requirements.txt
-
Run the main script:
python gui.py
-
Import image: Click the icon in the top left to select an image. File format is 32 x 32, white drawing on black background
-
Detect image: Click the detect button!