Object Detection
ONNX
English
dghs-realutils
dghs-imgutils
narugo1992 commited on
Commit
7469d41
·
verified ·
1 Parent(s): 747dac1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +31 -3
README.md CHANGED
@@ -1,6 +1,6 @@
1
  ---
2
  pipeline_tag: object-detection
3
- library_name: dghs-imgutils
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 |