simple-faster-rcnn-pytorch icon indicating copy to clipboard operation
simple-faster-rcnn-pytorch copied to clipboard

Why not use argmax of scores for label in suppress

Open tayden opened this issue 5 years ago • 0 comments
trafficstars

In the FasterRCNN.py module's _suppress function, All BG class scores are ignored and the FG box scores are kept as long as they are > 0.05. These then go into a class-wise NMS function.

Wouldn't it be better to first label all the boxes using the argmax of the probability scores, and then discard the BG boxes? It doesn't make sense to me to keep a box because it has a FG class confidence score of 0.1 if the BG class has a score of 0.9.

tayden avatar Dec 16 '19 22:12 tayden