Add NMS plugin in deepstream app

Accelerated Computing Intelligent Video Analytics DeepStream SDK
#1

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

#3

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

1 Like