Ghustwb / MobileNet-SSD-TensorRT

Accelerate mobileNet-ssd with tensorRT
188 stars 91 forks source link

one question i want to ask why is nms not used in this project. thank you in advance #12

Closed zhouleidcc closed 5 years ago

Ghustwb commented 5 years ago

NMS was used in detection_out layer,I used TensorRT's API , so you could not see it.

zhouleidcc commented 5 years ago

Thank you for your reply. But I am still confuse. What I concern is whether I need to apply NMS after getting boxes through API tensorNet.imageInference. Look forward to your reply, thanks again.