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