narugo1992
commited on
Update README.md
Browse files
README.md
CHANGED
|
@@ -15,6 +15,27 @@ These models are trained on [deepghs/anime_person_detection](https://huggingface
|
|
| 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 |
| person_detect_v0_l_yv11 | yolo | 87.3G | 25.3M | 0.79 | 0.359 | 0.84037 | 0.74055 | 0.82796 | 0.57272 | [plot](https://huggingface.co/deepghs/real_person_detection/blob/main/person_detect_v0_l_yv11/F1_curve.png) | [confusion](https://huggingface.co/deepghs/real_person_detection/blob/main/person_detect_v0_l_yv11/confusion_matrix_normalized.png) | `person` |
|
|
|
|
| 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_persons
|
| 26 |
+
|
| 27 |
+
print(detect_persons('yolo/solo.jpg'))
|
| 28 |
+
#[((0, 30, 398, 599), 'person', 0.926707923412323)]
|
| 29 |
+
print(detect_persons('yolo/2girls.jpg'))
|
| 30 |
+
# [((0, 74, 760, 1598), 'person', 0.7578195333480835), ((437, 33, 1200, 1600), 'person', 0.6875205039978027)]
|
| 31 |
+
print(detect_persons('yolo/3+cosplay.jpg'))
|
| 32 |
+
# [((106, 69, 347, 591), 'person', 0.8794167041778564), ((326, 14, 592, 534), 'person', 0.8018194437026978), ((167, 195, 676, 675), 'person', 0.5351650714874268)]
|
| 33 |
+
print(detect_persons('yolo/multiple.jpg'))
|
| 34 |
+
# [((1305, 441, 1891, 1534), 'person', 0.8789498805999756), ((206, 191, 932, 1533), 'person', 0.8423126935958862), ((1054, 170, 1417, 1055), 'person', 0.8138357996940613), ((697, 659, 1473, 1534), 'person', 0.7926754951477051), ((685, 247, 1128, 1526), 'person', 0.5261526703834534), ((690, 251, 1125, 1126), 'person', 0.4193646311759949)]
|
| 35 |
+
```
|
| 36 |
+
|
| 37 |
+
For more information, see [documentation of realutils](https://dghs-realutils.deepghs.org/main/api_doc/detect/person.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 |
| person_detect_v0_l_yv11 | yolo | 87.3G | 25.3M | 0.79 | 0.359 | 0.84037 | 0.74055 | 0.82796 | 0.57272 | [plot](https://huggingface.co/deepghs/real_person_detection/blob/main/person_detect_v0_l_yv11/F1_curve.png) | [confusion](https://huggingface.co/deepghs/real_person_detection/blob/main/person_detect_v0_l_yv11/confusion_matrix_normalized.png) | `person` |
|