Spaces:
Runtime error
Runtime error
Andrey
commited on
Commit
•
ca67f90
1
Parent(s):
2c16c94
Update descriptions. (#13)
Browse files- .github/README.md +11 -0
- pages/about.py +6 -3
- src/utils.py +2 -2
.github/README.md
CHANGED
@@ -1,2 +1,13 @@
|
|
1 |
![visitors](https://visitor-badge.glitch.me/badge?page_id=wissamantoun.arabicnlpapp)
|
2 |
[![DeepSource](https://static.deepsource.io/deepsource-badge-light-mini.svg)](https://deepsource.io/gh/Erlemar/digit-draw-detect/?ref=repository-badge )
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
![visitors](https://visitor-badge.glitch.me/badge?page_id=wissamantoun.arabicnlpapp)
|
2 |
[![DeepSource](https://static.deepsource.io/deepsource-badge-light-mini.svg)](https://deepsource.io/gh/Erlemar/digit-draw-detect/?ref=repository-badge )
|
3 |
+
|
4 |
+
This is a repo of my "Handwritten digit detector" pet-project. It uses a YOLOv3 model trained from scratch and Streamlit for frontent. You can see the live version of the app [here](https://huggingface.co/spaces/Artgor/digit-draw-detect).
|
5 |
+
|
6 |
+
If you are interested in reading more about this project, here are some links:
|
7 |
+
* [Project page on my personal website](https://andlukyane.com/project/drawn-digits-prediction)
|
8 |
+
* [A dataset with the digits and bounding boxes on Kaggle](https://www.kaggle.com/datasets/artgor/handwritten-digits-and-bounding-boxes)
|
9 |
+
* [Training code](https://github.com/Erlemar/pytorch_tempest_pet_)
|
10 |
+
* Blogpost on my personal website
|
11 |
+
* [Blogpost on medium](https://towardsdatascience.com/the-third-life-of-a-personal-pet-project-for-handwritten-digit-recognition-fd908dc8e7a1)
|
12 |
+
* Russian blogpost on habrahabr
|
13 |
+
* [W&B report](https://wandb.ai/al-3002-w/pet_project_object_detection/reports/Training-a-model-for-Handwritten-Object-Detection---VmlldzozMTgwMzA2?accessToken=yi6t4sz6iwr1yp78nfpvw71qao5wibak30np9tfft885tdj26g3tk91h1sie3h5m)
|
pages/about.py
CHANGED
@@ -6,8 +6,8 @@ The first idea of the project was conceived in the summer of 2017. I completed c
|
|
6 |
|
7 |
In 2019 I decided to update the [project](https://github.com/Erlemar/digit-draw-predict): I trained the new neural net in PyTorch and used cv2 to detect separate digits (people often drew multiple digits). More than that, the model had 11 classes - I made a separate class for "junk", as people often drew things for fun: animals, objects, or words.
|
8 |
|
9 |
-
The first two versions were deployed on Heroku's free plan, but in 2022 these plans were discontinued. I didn't want my project to die because of nostalgia, so I developed a new version and deployed it differently. The current version has an object detection model (yolo3 written from scratch) and 12 classes (digits, junk, and **censored**)
|
10 |
-
|
11 |
|
12 |
Initially, I considered deploying the app on Streamlit Cloud, but its computational limits were too low, so now the model is live on HuggingFace Spaces.
|
13 |
|
@@ -16,7 +16,10 @@ Initially, I considered deploying the app on Streamlit Cloud, but its computatio
|
|
16 |
* [Project page on my personal website](https://andlukyane.com/project/drawn-digits-prediction)
|
17 |
* [A dataset with the digits and bounding boxes on Kaggle](https://www.kaggle.com/datasets/artgor/handwritten-digits-and-bounding-boxes)
|
18 |
* [Training code](https://github.com/Erlemar/pytorch_tempest_pet_)
|
19 |
-
|
|
|
|
|
|
|
20 |
"""
|
21 |
|
22 |
st.markdown(text, unsafe_allow_html=True)
|
|
|
6 |
|
7 |
In 2019 I decided to update the [project](https://github.com/Erlemar/digit-draw-predict): I trained the new neural net in PyTorch and used cv2 to detect separate digits (people often drew multiple digits). More than that, the model had 11 classes - I made a separate class for "junk", as people often drew things for fun: animals, objects, or words.
|
8 |
|
9 |
+
The first two versions were deployed on Heroku's free plan, but in 2022 these plans were discontinued. I didn't want my project to die because of nostalgia, so I developed a new version and deployed it differently. The current version has an object detection model (yolo3 written from scratch) and 12 classes (digits, junk, and **censored**).
|
10 |
+
If you want to know what does **censored** means, just try to draw something 😉
|
11 |
|
12 |
Initially, I considered deploying the app on Streamlit Cloud, but its computational limits were too low, so now the model is live on HuggingFace Spaces.
|
13 |
|
|
|
16 |
* [Project page on my personal website](https://andlukyane.com/project/drawn-digits-prediction)
|
17 |
* [A dataset with the digits and bounding boxes on Kaggle](https://www.kaggle.com/datasets/artgor/handwritten-digits-and-bounding-boxes)
|
18 |
* [Training code](https://github.com/Erlemar/pytorch_tempest_pet_)
|
19 |
+
* Blogpost on my personal website
|
20 |
+
* [Blogpost on medium](https://towardsdatascience.com/the-third-life-of-a-personal-pet-project-for-handwritten-digit-recognition-fd908dc8e7a1)
|
21 |
+
* Russian blogpost on habrahabr
|
22 |
+
* [Project code on GitHub](https://github.com/Erlemar/digit-draw-detect)
|
23 |
"""
|
24 |
|
25 |
st.markdown(text, unsafe_allow_html=True)
|
src/utils.py
CHANGED
@@ -58,8 +58,8 @@ def plot_img_with_rects(
|
|
58 |
x = xc - w / 2
|
59 |
y = yc - h / 2
|
60 |
label = int(label)
|
61 |
-
label = label if label != 10 else '
|
62 |
-
label = label if label != 11 else '
|
63 |
rect = [x, y, x + w, y + h]
|
64 |
|
65 |
rect_ = patches.Rectangle(
|
|
|
58 |
x = xc - w / 2
|
59 |
y = yc - h / 2
|
60 |
label = int(label)
|
61 |
+
label = label if label != 10 else 'censored'
|
62 |
+
label = label if label != 11 else 'other'
|
63 |
rect = [x, y, x + w, y + h]
|
64 |
|
65 |
rect_ = patches.Rectangle(
|