AdamCodd commited on
Commit
5c8bd7e
·
verified ·
1 Parent(s): bb47330

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +50 -3
README.md CHANGED
@@ -1,3 +1,50 @@
1
- ---
2
- license: cc-by-nc-4.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: cc-by-nc-4.0
3
+ base_model: ultralytics/yolo11n
4
+ library_name: ultralytics
5
+ metrics:
6
+ - precision
7
+ - recall
8
+ - mAP
9
+ pipeline_tag: object-detection
10
+ tags:
11
+ - face-detection
12
+ - age-estimation
13
+ - yolo11
14
+ - onnx
15
+ ---
16
+
17
+ # YOLO11 Nano - Age Classification
18
+
19
+ This is a lightweight face detection and age classification model based on the **YOLO11 Nano** architecture. It was fine-tuned for 300 epochs on a heavily curated dataset of approximately 75,000 face images to classify individuals into three broad age brackets: **0-14**, **15-22**, and **22+**.
20
+
21
+ The training data was sourced and filtered from IMDB-Wiki, Adience, UTKFace, and FairFace. Consequently, this model is released under a **CC-BY-NC-4.0** license and is intended strictly for **non-commercial research**.
22
+
23
+ ## Performance
24
+
25
+ The model achieves high precision despite its small size, making it suitable for edge devices or local privacy-preserving inference.
26
+
27
+ * **mAP@50:** ~89.5%
28
+ * **mAP@50-95:** ~86.5%
29
+ * **Recall:** ~87%
30
+ * **Precision:** ~83%
31
+
32
+ ### Confusion Matrix
33
+
34
+ The table below shows the normalized prediction results. The columns represent the actual age (True Labels), and the rows represent the model's predictions.
35
+
36
+ | | **True: 0-14** | **True: 15-22** | **True: 22+** |
37
+ | :--- | :---: | :---: | :---: |
38
+ | **Pred: 0-14** | **0.91** | 0.07 | 0.01 |
39
+ | **Pred: 15-22** | 0.08 | **0.74** | 0.16 |
40
+ | **Pred: 22+** | 0.01 | 0.19 | **0.82** |
41
+
42
+ The model is <u>highly reliable at identifying children</u>, with **91% accuracy** and minimal confusion with adults (only 1% of children were predicted as 22+).
43
+
44
+ ## Limitations & Legal
45
+
46
+ Since this model was trained exclusively on real photos, performance may degrade on AI-generated faces, cartoons, or 3D renders. However, personal tests suggest it maintains reasonable detection capabilities on realistic AI generated content.
47
+
48
+ * **Accuracy:** This model is not 100% accurate. It uses a "buffer" class (15-22) to handle the ambiguity of young adulthood, but obviously it's not perfect.
49
+ * **Critical Use:** It should **not** be used for critical age verification or legal identity checks.
50
+ * **Biometric Privacy:** This model processes biometric features (faces). It is provided for research and local-inference contexts only and should not be deployed for mass surveillance or without considering local regulations (such as the GDPR guidelines in Europe or CCPA in the US).