narugo1992
commited on
Update README.md
Browse files
README.md
CHANGED
|
@@ -15,6 +15,27 @@ These models are trained on [deepghs/anime_head_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 |
| head_detect_v0_l_yv11 | yolo | 87.3G | 25.3M | 0.81 | 0.199 | 0.90226 | 0.72872 | 0.81049 | 0.5109 | [plot](https://huggingface.co/deepghs/real_head_detection/blob/main/head_detect_v0_l_yv11/F1_curve.png) | [confusion](https://huggingface.co/deepghs/real_head_detection/blob/main/head_detect_v0_l_yv11/confusion_matrix_normalized.png) | `head` |
|
|
|
|
| 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_heads
|
| 26 |
+
|
| 27 |
+
print(detect_heads('yolo/solo.jpg'))
|
| 28 |
+
# [((162, 47, 305, 210), 'head', 0.7701659202575684)]
|
| 29 |
+
print(detect_heads('yolo/2girls.jpg'))
|
| 30 |
+
# [((683, 48, 1199, 754), 'head', 0.8410779237747192), ((105, 91, 570, 734), 'head', 0.8339194059371948)]
|
| 31 |
+
print(detect_heads('yolo/3+cosplay.jpg'))
|
| 32 |
+
# [((329, 194, 426, 309), 'head', 0.8123012781143188), ((359, 20, 448, 122), 'head', 0.8047150373458862), ((185, 81, 265, 166), 'head', 0.7797152996063232)]
|
| 33 |
+
print(detect_heads('yolo/multiple.jpg'))
|
| 34 |
+
# [((867, 259, 1084, 527), 'head', 0.8264595866203308), ((1364, 448, 1583, 724), 'head', 0.8254891633987427), ((480, 201, 781, 565), 'head', 0.8191508054733276), ((1189, 175, 1398, 412), 'head', 0.8097156286239624), ((1028, 671, 1277, 992), 'head', 0.8084591627120972)]
|
| 35 |
+
```
|
| 36 |
+
|
| 37 |
+
For more information, see [documentation of realutils](https://dghs-realutils.deepghs.org/main/api_doc/detect/head.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 |
| head_detect_v0_l_yv11 | yolo | 87.3G | 25.3M | 0.81 | 0.199 | 0.90226 | 0.72872 | 0.81049 | 0.5109 | [plot](https://huggingface.co/deepghs/real_head_detection/blob/main/head_detect_v0_l_yv11/F1_curve.png) | [confusion](https://huggingface.co/deepghs/real_head_detection/blob/main/head_detect_v0_l_yv11/confusion_matrix_normalized.png) | `head` |
|