PreciseRoIPooling icon indicating copy to clipboard operation
PreciseRoIPooling copied to clipboard

IoU threshold in IoU Net

Open Muran337287 opened this issue 6 years ago • 2 comments

Hi, in your paper, predicted boxes with IoU larger than 0.5 will be fed into IoUNet. I have a question that did you have a try to set the threshold as 0.0? We will regard boxes with IoU>0.5 as positive boxes, but does IoU net have the ability to distinguish boxes with low IoU? If IoU net can only generate IoU score larger than 0.5, will it affect the performance?

Muran337287 avatar May 13 '19 07:05 Muran337287

No, definitely if all boxes during training are of IoU>0.5, you can not expect the model to generalize to low-IoU boxes. This threshold is important since using all bounding boxes to train the IoU head (boxes with arbitrary IoU) will affect the performance of the prediction accuracy. (simply because there are much more training examples with larger visual variance)

vacancy avatar Aug 21 '19 14:08 vacancy

I have some extra words about the thresholds we've tried at the bottom of #36. If you are interested that might be helpful.

jbr97 avatar Sep 17 '19 13:09 jbr97