yolov5_gpu_optimization
yolov5_gpu_optimization copied to clipboard
NMS GPU with Deepstream
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.
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