search

marcoslucianops / DeepStream-Yolo-Pose

NVIDIA DeepStream SDK 6.3 / 6.2 / 6.1.1 / 6.1 / 6.0.1 / 6.0 application for YOLO-Pose models
MIT License
109 stars 25 forks source link

Receiving an error while running demo on Jetson #3

Closed PrashantDixit0 closed 9 months ago

PrashantDixit0 commented 9 months ago

Traceback (most recent call last): File "deepstream.py", line 178, in tracker_src_pad_buffer_probe parse_pose_from_meta(frame_meta, obj_meta) File "deepstream.py", line 129, in parse_pose_from_meta x1 = int((data[(skeleton[i][0] - 1) * 3 + 0] - pad_x) / gain) IndexError: index 45 is out of bounds for axis 0 with size 0

marcoslucianops commented 9 months ago

Which model are you using?

PrashantDixit0 commented 9 months ago

Yolov8_Pose model downloaded and converted according to the official documentation given by ultralytics

marcoslucianops commented 9 months ago

https://github.com/marcoslucianops/DeepStream-Yolo-Pose/blob/master/docs/YOLOv8_Pose.md

PrashantDixit0 commented 9 months ago

For Current Config of DeepStream, Do I have to change size and shape of input image?. I think there the problem is occuring.

marcoslucianops commented 9 months ago

You need to use my export_yoloV8_pose.py file to export the model. Please follow the link I sent.

PrashantDixit0 commented 9 months ago

Thank you @marcoslucianops I'll follow export_yolov8_pose.py to export the model. Once I am able to run it, I'll close this Issue.

PrashantDixit0 commented 9 months ago

@marcoslucianops Just an update, It started running after exporting onnx file using export_yolov8_pose.py . Made it running at 19 fps 👍