About deepstream sdk with yolov5

Hello,


Can I use a python model trained with yolov5 network (eg best.pt) with deepstream sdk?

If you can, can you explain with an example?

Thank you.

Hi,

If you have the onnx type YOLOv5 model, it’s recommended to check if all the operations are supported in the TensorRT first.

$ /usr/src/tensorrt/bin/trtexec --onnx=[model]

If yes, it should be possible to deploy with Deepstream.
Here is a sample to run YOLOv4 with Deepstream for your reference:

Thanks.

1 Like