Views
No views yet
1from ezonnx import RTMDet
2
3rtm = RTMDet("s-coco") # you can choose "n-person","m-person","n-hand","tiny-coco","s-coco","m-coco","l-coco","x-coco"
4result = rtm("image.jpg")
5
6print(result.boxes) # (N, 4) array of bounding boxes
7print(result.classes) # (N,) array of class labels
8print(result.scores) # (N,) array of confidence scores
9print(result.visualized_img) # (H, W, 3) image with boxes