yolov5_gpu_optimization icon indicating copy to clipboard operation
yolov5_gpu_optimization copied to clipboard

NMS GPU with Deepstream

Open Arpit1496 opened this issue 2 years ago • 1 comments

I want to run NMS on GPU but with deepstream and not trt.py file you mentioned. I created the onnx engine using the steps as told in NMS on GPU and then tried to run that engine but it showed similar performance to that of CPU onnx engine (20fps for yolov5s6 on jetson nx 8 GB).

How can I run NMS on GPU while using deepstream-app.

Arpit1496 avatar Feb 19 '23 10:02 Arpit1496

If you want to run GPU NMS in DeepStream, you could follow the following link: https://github.com/NVIDIA-AI-IOT/yolov5_gpu_optimization#deepstream-deployment https://github.com/NVIDIA-AI-IOT/deepstream_tao_apps

Tyler-D avatar Feb 20 '23 06:02 Tyler-D