wuhp commited on
Commit
c82cece
·
verified ·
1 Parent(s): 8a9bdd8

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
+ # face2
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.99373
15
+
16
+ ## Class IDs
17
+
18
+ | Class ID | Class Name |
19
+ |----------|------------|
20
+ | 0 | Face |
21
+
22
+
23
+ ## Datasets Used
24
+
25
+ - face-detection-mik1i_v24
26
+ - faces-bfigz_v3
27
+ - head-qug6h_v2
28
+
29
+ ## Class Image Counts
30
+
31
+ | Class Name | Image Count |
32
+ |------------|-------------|
33
+ | Face | 15349 |
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('face2.pt')
49
+
50
+ # Perform inference on an image
51
+ results = model('path_to_image.jpg')
52
+
53
+ # Display results
54
+ results.show()