how to turn on nms in run_ssd_batch.py

[ynjiun]

I run run_ssd_batch.py and found there are a lot of overlapped bounding boxes output with different confidence level. I assume that nms should remove the lower confidence overlapped bounding boxes and only keep the highest confidence bounding boxes. I wonder the default nms is not turn on and could anyone advise how to turn on nms? and also what does it mean nms="hard" and "soft"? Thanks a lot for your help.

[ynjiun]

figured out why "a lot of overlapped bounding boxes output": this is because the nms algorithms only apply to the same class_index in the vision/ssd/predictor.py. If different class_index with overlap greater than iou_threshold, then it would not be be suppressed thus "a lot of overlapped bounding boxes" show up with different class labels even the highest probability one is correct label. Well, both the predictor and nms need to be revised to suppress across all classes.

[lweicker]

Hello,

Can you please tell me where did you find run_ssd_batch.py file?