Yolov5 giving unexpected outputs

Please provide complete information as applicable to your setup.

• Hardware Platform (Jetson / GPU) -> aws T4
• DeepStream Version -> 5.0
• TensorRT Version - 7.0
• NVIDIA GPU Driver Version (valid for GPU only) -> 440.82

@ersheng Followed your yolov4 repo to make TRT engine for yolov5 which was built successfully. Compared the output with pytorch mode and they are both same. But when I hook it up in DeepStream I am not getting any boxes. I have uploaded the code and relevant files here . Let me know if you have any pointers!



Sorry for late reply.
How is status of your YOLOv5 deployment to DeepStream now?

@ersheng Haven’t been able to make progress, tried with pytorch -> onnx -> tensorrt -> deepstream. But that too gives incorrect outputs. Precisely, too may bounding boxes just as in the case of when using tensorrtx/yolov5 to create deepstream app.


I’ve given somebody who is also handling YOLOv5 some hints here. Hopefully they are also helpful for you.

Here’s a great tuto : GitHub - marcoslucianops/DeepStream-Yolo: NVIDIA DeepStream SDK 5.1 configuration for YOLO models

1 Like