search

triple-Mu / YOLOv8-TensorRT

YOLOv8 using TensorRT accelerate !
MIT License
1.32k stars 228 forks source link

pose_postprocess for multiple classes #205

Open saim212 opened 6 months ago

saim212 commented 6 months ago

I trained the pose model with 2 classes. Each class has 5 keypoints. I successfully converted the model, but I'm unable to extract the multiple classes from the output. How I update the pose_postprocess for the multiple classes? (python)

triple-Mu commented 4 months ago

I trained the pose model with 2 classes. Each class has 5 keypoints. I successfully converted the model, but I'm unable to extract the multiple classes from the output. How I update the pose_postprocess for the multiple classes? (python)

Not support yet. Because the coco pose just one class. Could you please give me a pt model for testing?

saim212 commented 4 months ago

I resolved this issue using ultralytics pose post-processing. I can't provide you the current model. However, I can prepare a sample model, if it is still required for testing.

triple-Mu commented 3 months ago

I resolved this issue using ultralytics pose post-processing. I can't provide you the current model. However, I can prepare a sample model, if it is still required for testing.

Sure I will really appreciate.