narugo1992
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
---
|
2 |
pipeline_tag: object-detection
|
3 |
-
library_name: dghs-
|
4 |
datasets:
|
5 |
- HuggingFaceM4/COCO
|
6 |
language:
|
@@ -8,8 +8,37 @@ language:
|
|
8 |
metrics:
|
9 |
- precision
|
10 |
- recall
|
|
|
|
|
|
|
|
|
11 |
---
|
12 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
13 |
| Model | Type | FLOPS | Params | Labels | precision(B) | recall(B) | mAP50(B) | mAP50-95(B) |
|
14 |
|:--------:|:------:|:-------:|:--------:|:--------------------------------------------------------------------------------------------------------------------------------------------------:|:--------------:|:-----------:|:----------:|:-------------:|
|
15 |
| rtdetr-x | rtdetr | 233G | 67.5M | [`person`, `bicycle`, `car`, `motorcycle`, `airplane` ... 80 labels in total](https://huggingface.co/deepghs/yolos/blob/main/rtdetr-x/labels.json) | N/A | N/A | N/A | N/A |
|
@@ -39,5 +68,4 @@ metrics:
|
|
39 |
| yolov5lu | yolo | 136G | 53.2M | [`person`, `bicycle`, `car`, `motorcycle`, `airplane` ... 80 labels in total](https://huggingface.co/deepghs/yolos/blob/main/yolov5lu/labels.json) | N/A | N/A | N/A | N/A |
|
40 |
| yolov5mu | yolo | 64.6G | 25.1M | [`person`, `bicycle`, `car`, `motorcycle`, `airplane` ... 80 labels in total](https://huggingface.co/deepghs/yolos/blob/main/yolov5mu/labels.json) | N/A | N/A | N/A | N/A |
|
41 |
| yolov5su | yolo | 24.2G | 9.15M | [`person`, `bicycle`, `car`, `motorcycle`, `airplane` ... 80 labels in total](https://huggingface.co/deepghs/yolos/blob/main/yolov5su/labels.json) | N/A | N/A | N/A | N/A |
|
42 |
-
| yolov5nu | yolo | 7.84G | 2.65M | [`person`, `bicycle`, `car`, `motorcycle`, `airplane` ... 80 labels in total](https://huggingface.co/deepghs/yolos/blob/main/yolov5nu/labels.json) | N/A | N/A | N/A | N/A |
|
43 |
-
|
|
|
1 |
---
|
2 |
pipeline_tag: object-detection
|
3 |
+
library_name: dghs-realutils
|
4 |
datasets:
|
5 |
- HuggingFaceM4/COCO
|
6 |
language:
|
|
|
8 |
metrics:
|
9 |
- precision
|
10 |
- recall
|
11 |
+
tags:
|
12 |
+
- onnx
|
13 |
+
- dghs-imgutils
|
14 |
+
- dghs-realutils
|
15 |
---
|
16 |
|
17 |
+
Official pretrained models from [ultralytics](https://github.com/ultralytics/ultralytics).
|
18 |
+
|
19 |
+
Use this with `dghs-realutils`
|
20 |
+
|
21 |
+
```shell
|
22 |
+
pip install dghs-realutils
|
23 |
+
```
|
24 |
+
|
25 |
+
```python
|
26 |
+
from realutils.detect import detect_by_yolo
|
27 |
+
|
28 |
+
print(detect_by_yolo('yolo/unsplash_aJafJ0sLo6o.jpg'))
|
29 |
+
# [((450, 317, 567, 599), 'person', 0.9004617929458618)]
|
30 |
+
print(detect_by_yolo('yolo/unsplash_n4qQGOBgI7U.jpg'))
|
31 |
+
# [((73, 101, 365, 409), 'vase', 0.9098997116088867), ((441, 215, 659, 428), 'vase', 0.622944176197052), ((5, 1, 428, 377), 'potted plant', 0.5178268551826477)]
|
32 |
+
print(detect_by_yolo('yolo/unsplash_vUNQaTtZeOo.jpg'))
|
33 |
+
# [((381, 103, 676, 448), 'bird', 0.9061452150344849)]
|
34 |
+
print(detect_by_yolo('yolo/unsplash_YZOqXWF_9pk.jpg'))
|
35 |
+
# [((315, 100, 690, 532), 'horse', 0.9453459978103638), ((198, 181, 291, 256), 'horse', 0.917123556137085), ((145, 173, 180, 249), 'horse', 0.7972317337989807), ((660, 138, 701, 170), 'horse', 0.4843617379665375)]
|
36 |
+
```
|
37 |
+
|
38 |
+
For more information, see [documentation of realutils](https://dghs-realutils.deepghs.org/main/api_doc/detect/yolo.html).
|
39 |
+
|
40 |
+
## Models Available
|
41 |
+
|
42 |
| Model | Type | FLOPS | Params | Labels | precision(B) | recall(B) | mAP50(B) | mAP50-95(B) |
|
43 |
|:--------:|:------:|:-------:|:--------:|:--------------------------------------------------------------------------------------------------------------------------------------------------:|:--------------:|:-----------:|:----------:|:-------------:|
|
44 |
| rtdetr-x | rtdetr | 233G | 67.5M | [`person`, `bicycle`, `car`, `motorcycle`, `airplane` ... 80 labels in total](https://huggingface.co/deepghs/yolos/blob/main/rtdetr-x/labels.json) | N/A | N/A | N/A | N/A |
|
|
|
68 |
| yolov5lu | yolo | 136G | 53.2M | [`person`, `bicycle`, `car`, `motorcycle`, `airplane` ... 80 labels in total](https://huggingface.co/deepghs/yolos/blob/main/yolov5lu/labels.json) | N/A | N/A | N/A | N/A |
|
69 |
| yolov5mu | yolo | 64.6G | 25.1M | [`person`, `bicycle`, `car`, `motorcycle`, `airplane` ... 80 labels in total](https://huggingface.co/deepghs/yolos/blob/main/yolov5mu/labels.json) | N/A | N/A | N/A | N/A |
|
70 |
| yolov5su | yolo | 24.2G | 9.15M | [`person`, `bicycle`, `car`, `motorcycle`, `airplane` ... 80 labels in total](https://huggingface.co/deepghs/yolos/blob/main/yolov5su/labels.json) | N/A | N/A | N/A | N/A |
|
71 |
+
| yolov5nu | yolo | 7.84G | 2.65M | [`person`, `bicycle`, `car`, `motorcycle`, `airplane` ... 80 labels in total](https://huggingface.co/deepghs/yolos/blob/main/yolov5nu/labels.json) | N/A | N/A | N/A | N/A |
|
|