dp03 commited on
Commit
812fe32
·
verified ·
1 Parent(s): c720b8b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -9
README.md CHANGED
@@ -1,17 +1,17 @@
1
 
2
  # Model Description
3
- This model detects dogs in images and catorgizes them into small, medium, and large based on average weight of an adult of that breed. The weight classes generall follow: less than 30lbs for small doges, between 30lbs and 50lbs for medium and greater than 50lbs for large dogs.
4
- Training was done by fine tunning the yolov11 model. Due to the large physical differences between dog breed, this model is intended to be used to determine counts of each type in order to better meet the needs to the group in the area.
5
  ***
6
  # Training Data
7
- This model is trained using the follow roboflow dataset: [Link](https://universe.roboflow.com/igor-romanica-gmail-com/stanford-dogs-0pff9).
8
- The roboflow page is using a subset of the [Standford Dogs Dataset](http://vision.stanford.edu/aditya86/ImageNetDogs/). The subset consits of images of 60 breeds across 9884 images, about half the breed and image count of the orginal dataset.
9
- Annotations included manually sorting each of the 60 breeds into a catagory based on weight(as detailed above). Additionally, some classes were deleted due to the large wieght ranges of the breed. For example, [Xoloitzcuintle](https://en.wikipedia.org/wiki/Xoloitzcuintle) are usually broken into three sub-breeds with different sizes but they are labeled in the dataset under one cateogry.
10
 
11
  ### *Class Breakdown*
12
  | Metric | Small | Medium | Large |
13
  |--------|--------|----------|------ |
14
- | Percent| 39% | 37% | 24% |
15
  | Count | 4,058 | 3,860 | 2,524 |
16
 
17
  ### *Training Split*
@@ -25,7 +25,7 @@ Annotations included manually sorting each of the 60 breeds into a catagory base
25
  * Trained on Google Collab using A100
26
  * Limited to 200 epochs and 100 patience
27
  * Ran for 73 epochs, best at 63
28
- * 4.9 Hours training on ~24k images
29
  * 10k base
30
  * 14k augmented on exposure and blur
31
  ***
@@ -47,13 +47,14 @@ Less than a .04 difference between classes for each metric.
47
  ### *Visualizations*
48
  <img alt="Confusion Matrix" src="https://huggingface.co/cvtechniques/DogTypeDetection/resolve/main/confusion_matrix_normalized.png" width="700"></img>
49
  <img alt="F1-Confidence Graph" src="https://huggingface.co/cvtechniques/DogTypeDetection/resolve/main/BoxF1_curve.png" width="700"></img>
50
- <img alt="Precsiosn-Confidence Graph" src="https://huggingface.co/cvtechniques/DogTypeDetection/resolve/main/BoxP_curve.png"></img>
51
  ### *Performance Analysis*
52
 
53
- This model had high metrics across each of the classes, meeting the success threshold in precision, recall and F1 score. The confusion matrics shows some slight over guessing, as each of the classes had a 25% to 40% rates of being prediceted when that area was actual background. The model also predicted small dogs as large dogs 10% of the time, which was right at the limit set before training. That being said, the matrix still has high values of 80%-85% along the true postive diagonal. The 100% precioson peak at
54
  ***
55
  # Limitations and Biases
56
  ### *Known failure cases*
 
57
  ### *Poor performing classes*
58
  ### *Data biases*
59
  ### *Environmental/contextual limitations*
 
1
 
2
  # Model Description
3
+ This model detects dogs in images and categorizes them into small, medium, and large based on the average weight of an adult of that breed. The weight classes generally follow: less than 30lbs for small dogs, between 30 lbs and 50lbs for medium dogs, and greater than 50lbs for large dogs.
4
+ Training was done by fine-tuning the YOLOv11 model. Due to the large physical differences between dog breeds, this model is intended to be used to determine counts of each type to better meet the needs of the group in the area.
5
  ***
6
  # Training Data
7
+ This model is trained using the following Roboflow dataset: [Link](https://universe.roboflow.com/igor-romanica-gmail-com/stanford-dogs-0pff9).
8
+ The Roboflow page is using a subset of the [Stanford Dogs Dataset](http://vision.stanford.edu/aditya86/ImageNetDogs/). The subset consists of images of 60 breeds across 9884 images, about half the breed and image count of the original dataset.
9
+ Annotations included manually sorting each of the 60 breeds into a category based on weight (as detailed above). Additionally, some classes were deleted due to the large weight ranges of the breed. For example, [Xoloitzcuintles](https://en.wikipedia.org/wiki/Xoloitzcuintle) are usually broken into three sub-breeds with different sizes, but they are labeled in the dataset under one category.
10
 
11
  ### *Class Breakdown*
12
  | Metric | Small | Medium | Large |
13
  |--------|--------|----------|------ |
14
+ | Percent | 39% | 37% | 24% |
15
  | Count | 4,058 | 3,860 | 2,524 |
16
 
17
  ### *Training Split*
 
25
  * Trained on Google Collab using A100
26
  * Limited to 200 epochs and 100 patience
27
  * Ran for 73 epochs, best at 63
28
+ * 4.9 hours of training on ~24k images
29
  * 10k base
30
  * 14k augmented on exposure and blur
31
  ***
 
47
  ### *Visualizations*
48
  <img alt="Confusion Matrix" src="https://huggingface.co/cvtechniques/DogTypeDetection/resolve/main/confusion_matrix_normalized.png" width="700"></img>
49
  <img alt="F1-Confidence Graph" src="https://huggingface.co/cvtechniques/DogTypeDetection/resolve/main/BoxF1_curve.png" width="700"></img>
50
+ <img alt="Precsiosn-Confidence Graph" src="https://huggingface.co/cvtechniques/DogTypeDetection/resolve/main/BoxP_curve.png" width="700"></img>
51
  ### *Performance Analysis*
52
 
53
+ This model had high metrics across each of the classes, meeting the success threshold in precision, recall, and F1 score. The confusion matrix shows some slight overguessing, as each of the classes had a 25% to 40% rate of being predicted when that area was actual background. The model also predicted small dogs as large dogs 10% of the time, which was right at the limit set before training. That being said, the matrix still has high values of 80%-85% along the true positive diagonal. The 100% precision peak at 100% confidence does raise some red flags. This is addressed in the *Known failure cases* section.
54
  ***
55
  # Limitations and Biases
56
  ### *Known failure cases*
57
+
58
  ### *Poor performing classes*
59
  ### *Data biases*
60
  ### *Environmental/contextual limitations*