DeepStream-Yolo icon indicating copy to clipboard operation
DeepStream-Yolo copied to clipboard

GPU based nms

Open bug-developer021 opened this issue 1 year ago • 1 comments

Hi! you did greate job! And there was a question which confused me. According to New optimized NMS, you changed cluster-mode from 4 to 2. Does it means that you used deepstream nms function rather than customed nms? As far as i know, deepstream nms funciton is cpu based.

bug-developer021 avatar Sep 14 '22 12:09 bug-developer021

Yes, I reverted the NMS to CPU due to accuracy drop. I will start new tests with GPU NMS and I will update the repo when I get good results again.

marcoslucianops avatar Sep 23 '22 13:09 marcoslucianops