uralman commited on
Commit
1d98345
·
verified ·
1 Parent(s): 54ecba0

Add model card for YOLO26l WiderFace fine-tune

Browse files
Files changed (1) hide show
  1. README.md +98 -0
README.md ADDED
@@ -0,0 +1,98 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language:
3
+ - en
4
+ license: agpl-3.0
5
+ library_name: ultralytics
6
+ tags:
7
+ - ultralytics
8
+ - yolo
9
+ - yolo26
10
+ - object-detection
11
+ - face-detection
12
+ - widerface
13
+ pipeline_tag: object-detection
14
+ ---
15
+
16
+ # YOLO26l face detector (WiderFace fine-tune)
17
+
18
+ Single-class **face** detector based on **[Ultralytics YOLO26l](https://docs.ultralytics.com/models/yolo26/)**, fine-tuned on **[WIDER FACE](http://shuoyang1213.me/WIDERFACE/)**.
19
+
20
+ This repo contains the Ultralytics `best.pt` checkpoint from a supervised fine-tune of the official YOLO26 large detection model.
21
+
22
+ > Base model: [YOLO26 documentation](https://docs.ultralytics.com/models/yolo26/) · [Ultralytics GitHub](https://github.com/ultralytics/ultralytics)
23
+ > Dataset: [WIDER FACE](http://shuoyang1213.me/WIDERFACE/)
24
+
25
+ ## Files
26
+
27
+ | File | Description |
28
+ |------|-------------|
29
+ | `best.pt` | Ultralytics YOLO weights (`ultralytics.YOLO`) |
30
+ | `metrics.txt` | Snapshot of peak validation metrics from this run |
31
+
32
+ ## Quick start
33
+
34
+ ```bash
35
+ pip install ultralytics
36
+ huggingface-cli download uralman/yolo26l-widerface best.pt --local-dir .
37
+ ```
38
+
39
+ ```python
40
+ from ultralytics import YOLO
41
+
42
+ model = YOLO("best.pt")
43
+ results = model.predict("image.jpg", imgsz=1280, conf=0.25)
44
+ ```
45
+
46
+ Recommended inference size: **1280** (same as training).
47
+
48
+ ## Training summary
49
+
50
+ | Item | Value |
51
+ |------|-------|
52
+ | Base weights | Ultralytics `yolo26l.pt` |
53
+ | Task | `detect`, single class `face` |
54
+ | Dataset | WIDER FACE converted to YOLO format (~12.9k train / ~3.2k val) |
55
+ | Image size | 1280 |
56
+ | Hardware | 2× NVIDIA H100, DDP |
57
+ | Batch | 8, AMP, `cache=ram` |
58
+ | Schedule | up to 200 epochs, patience 50, cosine LR |
59
+ | Augmentations | mosaic, multi-scale, Ultralytics detect defaults |
60
+ | Stopped at | epoch **73** (early stop; gains near plateau) |
61
+
62
+ ## Validation metrics (WIDER FACE val)
63
+
64
+ Ultralytics built-in box metrics on the YOLO-formatted WIDER FACE validation split used for this run. Peak selected by **mAP50-95**:
65
+
66
+ | Metric | Epoch 1 | Best (epoch 71) |
67
+ |--------|--------:|-------------------------------:|
68
+ | Precision | — | 0.896 |
69
+ | Recall | — | 0.725 |
70
+ | mAP50 | 0.699 | **0.801** |
71
+ | mAP50-95 | 0.331 | **0.455** |
72
+
73
+ These are COCO-style Ultralytics mAP scores for this training setup — **not** the official WIDER FACE Easy/Medium/Hard evaluation protocol. Useful for reproducing this run; absolute numbers depend on label conversion, `imgsz`, and NMS.
74
+
75
+ ## Experiment notes (ClearML)
76
+
77
+ Training was logged in ClearML (project `yolo-face`). Highlights from the run:
78
+
79
+ - Val **mAP50** rose from ~0.70 → ~0.80 and **mAP50-95** from ~0.33 → ~0.455 over ~70 epochs
80
+ - Precision / recall at best checkpoint ≈ **0.90 / 0.72**
81
+ - Stable epoch time ≈ **6.5–7 min** on 2×H100 at batch 8 / imgsz 1280
82
+ - Train & val box/cls losses plus per-epoch detection metrics were tracked throughout; DDP job with additional train-batch scalars for monitoring
83
+
84
+ ## Intended use
85
+
86
+ - Face **detection** (boxes) for privacy blur or similar downstream pipelines
87
+ - Benchmarking YOLO26-scale face detectors on WIDER FACE-style data
88
+
89
+ ## Limitations
90
+
91
+ - Domain shift is expected outside WIDER FACE-like photos (extreme motion blur, unusual viewpoints, etc.)
92
+ - Not a face recognition / re-identification model
93
+ - Ultralytics stack is **AGPL-3.0** — review [license terms](https://www.ultralytics.com/license) before commercial redistribution
94
+
95
+ ## Acknowledgements
96
+
97
+ - [Ultralytics YOLO26](https://docs.ultralytics.com/models/yolo26/)
98
+ - [WIDER FACE](http://shuoyang1213.me/WIDERFACE/) (Yang et al.)