marcoslucianops / DeepStream-Yolo

NVIDIA DeepStream SDK 7.0 / 6.4 / 6.3 / 6.2 / 6.1.1 / 6.1 / 6.0.1 / 6.0 / 5.1 implementation for YOLO models
MIT License
1.38k stars 343 forks source link

Yolov5s is not giving accurate detections after converting to onnx #536

Open vmukund36 opened 1 month ago

vmukund36 commented 1 month ago

@marcoslucianops I have converted my custom .pt model to .onnx using the export_yolov5.py file but when I run inference on deepstream(6.3) using Jetson AGX Orin, I get wrong detections and I dont get the correct ones also properly. If I infer it using Ultralytics, I am getting accurate results but not with deepstream. What could be the reason?