Update README.md
Browse files
README.md
CHANGED
@@ -15,6 +15,27 @@ These models are trained on [deepghs/anime_face_detection](https://huggingface.c
|
|
15 |
|
16 |
So both anime and photos are supported.
|
17 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
18 |
| Model | Type | FLOPS | Params | F1 Score | Threshold | precision(B) | recall(B) | mAP50(B) | mAP50-95(B) | F1 Plot | Confusion | Labels |
|
19 |
|:---------------------:|:------:|:-------:|:--------:|:----------:|:-----------:|:--------------:|:-----------:|:----------:|:-------------:|:-------------------------------------------------------------------------------------------------------:|:-------------------------------------------------------------------------------------------------------------------------------:|:--------:|
|
20 |
| face_detect_v0_l_yv11 | yolo | 87.3G | 25.3M | 0.77 | 0.291 | 0.88458 | 0.67474 | 0.76666 | 0.45722 | [plot](https://huggingface.co/deepghs/real_face_detection/blob/main/face_detect_v0_l_yv11/F1_curve.png) | [confusion](https://huggingface.co/deepghs/real_face_detection/blob/main/face_detect_v0_l_yv11/confusion_matrix_normalized.png) | `face` |
|
|
|
15 |
|
16 |
So both anime and photos are supported.
|
17 |
|
18 |
+
Use this with `dghs-realutils`
|
19 |
+
|
20 |
+
```shell
|
21 |
+
pip install dghs-realutils
|
22 |
+
```
|
23 |
+
|
24 |
+
```python
|
25 |
+
from realutils.detect import detect_faces
|
26 |
+
|
27 |
+
print(detect_faces('yolo/solo.jpg'))
|
28 |
+
# [((157, 94, 252, 208), 'face', 0.8836570382118225)]
|
29 |
+
print(detect_faces('yolo/2girls.jpg'))
|
30 |
+
# [((718, 154, 1110, 728), 'face', 0.8841166496276855), ((157, 275, 519, 715), 'face', 0.8668240904808044)]
|
31 |
+
print(detect_faces('yolo/3+cosplay.jpg'))
|
32 |
+
# [((349, 227, 413, 305), 'face', 0.8543888330459595), ((383, 61, 432, 117), 'face', 0.8080574870109558), ((194, 107, 245, 162), 'face', 0.8035706877708435)]
|
33 |
+
print(detect_faces('yolo/multiple.jpg'))
|
34 |
+
# [((1070, 728, 1259, 985), 'face', 0.8765808939933777), ((548, 286, 760, 558), 'face', 0.8693087697029114), ((896, 315, 1067, 520), 'face', 0.8671919107437134), ((1198, 220, 1342, 406), 'face', 0.8485829830169678), ((1376, 526, 1546, 719), 'face', 0.8469308018684387)]
|
35 |
+
```
|
36 |
+
|
37 |
+
For more information, see [documentation of realutils](https://dghs-realutils.deepghs.org/main/api_doc/detect/face.html).
|
38 |
+
|
39 |
| Model | Type | FLOPS | Params | F1 Score | Threshold | precision(B) | recall(B) | mAP50(B) | mAP50-95(B) | F1 Plot | Confusion | Labels |
|
40 |
|:---------------------:|:------:|:-------:|:--------:|:----------:|:-----------:|:--------------:|:-----------:|:----------:|:-------------:|:-------------------------------------------------------------------------------------------------------:|:-------------------------------------------------------------------------------------------------------------------------------:|:--------:|
|
41 |
| face_detect_v0_l_yv11 | yolo | 87.3G | 25.3M | 0.77 | 0.291 | 0.88458 | 0.67474 | 0.76666 | 0.45722 | [plot](https://huggingface.co/deepghs/real_face_detection/blob/main/face_detect_v0_l_yv11/F1_curve.png) | [confusion](https://huggingface.co/deepghs/real_face_detection/blob/main/face_detect_v0_l_yv11/confusion_matrix_normalized.png) | `face` |
|