A progressive web application for drawing that automatically corrects hand-drawn circles and rectangles with the power of YOLOv5.
Check it out at https://illright.github.io/draw-ill-help.
- 🏎️ Completely serverless
- 🌐 Works offline
- 🧠 Runs YOLOv5 Nano in the background
- 🗃️ Built-in dataset generator
- 😎 Automatic dark mode
- 🍰 Follows the principles of Feature-Sliced Design
| Home page | Drawing canvas | Dataset generator |
|---|---|---|
|
|
|
|
Since the selected model is the smallest in the family of YOLOv5, it has very small inference times and file size. However, that also has an impact on its predictive capabilities.
See the charts below for the main performance characteristics of the model during training. The model had trained for around 350 epochs, early stopping with a patience value of 100 epochs due to no improvement at the target metric (0.9 of mAP (0.5:0.95) and 0.1 of mAP (0.5)). The horizontal axis represents the epochs (aka steps).
| mAP (0.5) | mAP (0.5:0.95) | Precision | Recall |
|---|---|---|---|
|
|
|
|
Grab the model.zip file from the latest release and unpack its contents into static/. Then run the usual commands:
pnpm i
pnpm devThe Jupyter notebook with explanations for the training process is available in the yolov5/ directory.
The recommended way to run the notebook is with Google Colaboratory. If you're using the GPU session and the free Colab account, the dataset of around 1.5k 416x416 images should take around two hours to train.
The source code of this project is distributed under the terms of the MIT license. Click here to learn what that means.