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 |
|
|
|