YOLOv8-TensorRT icon indicating copy to clipboard operation
YOLOv8-TensorRT copied to clipboard

issue related to detect multiple objects using the pose model

Open xcn700418 opened this issue 9 months ago • 0 comments

I successfully convert the model to onnx format and build the tensorRT engine. However, I am using the infer-pose.py to detect the object bounding boxes, each with associated keypoints, there are suppose to be 3 bounding boxes, but I only got 1 after nms. I check the coordinate of the detections that has higher scores than the conf_thres, they are quite similar, which means the model did not successfully detect the other object, but when i use the .pt model for inference, it did.

May I ask how can I modify the code to achieve multiple object detection? Is it related to the model conversion to onnx, or the tensorRT engine building stage, or related to the pose_postprocess function?

xcn700418 avatar Mar 18 '25 02:03 xcn700418