Add NMS plugin in deepstream app

Please provide complete information as applicable to your setup.

• dGPU aws T4
• DeepStream Version -> 4.0.2_x86_64
• TensorRT Version -> 6.0
• NVIDIA GPU Driver Version (valid for GPU only) -> 440.82

I was trying to build a DeepStream app for RetinaFace with the help of deepstream-test3. The TensorRT engine fie that I am using does not have NMS(Non max Suppression). Is there a way to do NMS in DeepStream app after inference from model with the help of some sort of plugin maybe if yes, please help me out.

Thanks in advance

You can refer to sample sources/objectDetector_SSD/ which have NMS parsing support

1 Like