bukuroo commited on
Commit
2dc8335
·
verified ·
1 Parent(s): 0e1490e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +25 -3
README.md CHANGED
@@ -1,3 +1,25 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - onnx
5
+ - object-detection
6
+ - ezonnx
7
+ ---
8
+
9
+ RTMDet real-time object detection model.
10
+ [MMPose](https://github.com/open-mmlab/mmpose/tree/main/projects/rtmpose)
11
+
12
+ ```python
13
+ from ezonnx import RTMDet
14
+
15
+ rtm = RTMDet("s-coco") # you can choose "n-person","m-person","n-hand","tiny-coco","s-coco","m-coco","l-coco","x-coco"
16
+ result = rtm("image.jpg")
17
+
18
+ print(result.boxes) # (N, 4) array of bounding boxes
19
+ print(result.classes) # (N,) array of class labels
20
+ print(result.scores) # (N,) array of confidence scores
21
+ print(result.visualized_img) # (H, W, 3) image with boxes
22
+ ```
23
+
24
+
25
+