wuhp commited on
Commit
c902ae2
·
verified ·
1 Parent(s): ba9b49a

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +54 -0
README.md ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # dog-75e-11n
2
+
3
+ ## Model Overview
4
+
5
+ **Architecture:** YOLOv11
6
+ **Training Epochs:** 75
7
+ **Batch Size:** 32
8
+ **Optimizer:** auto
9
+ **Learning Rate:** 0.0005
10
+ **Data Augmentation Level:** Moderate
11
+
12
+ ## Training Metrics
13
+
14
+ - **[email protected]:** 0.99457
15
+
16
+ ## Class IDs
17
+
18
+ | Class ID | Class Name |
19
+ |----------|------------|
20
+ | 0 | Full |
21
+ | 1 | Head |
22
+
23
+
24
+ ## Datasets Used
25
+
26
+ - doggo-detection-assignment-2_v2
27
+
28
+ ## Class Image Counts
29
+
30
+ | Class Name | Image Count |
31
+ |------------|-------------|
32
+ | Full | 27965 |
33
+ | Head | 27959 |
34
+
35
+
36
+ ## Description
37
+
38
+ This model was trained using the YOLOv11 architecture on a custom dataset. The training process involved 75 epochs with a batch size of 32. The optimizer used was **auto** with an initial learning rate of 0.0005. Data augmentation was set to the **Moderate** level to enhance model robustness.
39
+
40
+ ## Usage
41
+
42
+ To use this model for inference, follow the instructions below:
43
+
44
+ ```python
45
+ from ultralytics import YOLO
46
+
47
+ # Load the trained model
48
+ model = YOLO('best.pt')
49
+
50
+ # Perform inference on an image
51
+ results = model('path_to_image.jpg')
52
+
53
+ # Display results
54
+ results.show()