Andrey commited on
Commit
ca67f90
1 Parent(s): 2c16c94

Update descriptions. (#13)

Browse files
Files changed (3) hide show
  1. .github/README.md +11 -0
  2. pages/about.py +6 -3
  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
- . 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,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 'penis'
62
- label = label if label != 11 else 'junk'
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(