在这里插一句,在R-CNN中用到IoU阈值的有两个地方,分别是Training时Positive与Negative判定,和Inference时计算mAP。论文中强调的IoU阈值指的是Training时Positive和Negative判定处。
这说的不清楚,简单说一下从RPN到FAST-RCNN的过程: 1、RPN是anchorbased,一共会有20000个左右的anchor;
2、前向的时候,RPN会基于anchor,产生20000个左右的proposal;
3、通过anchor和gt的IOU,选出大于0.7(或者和GT框IOU最大的anchor)的128个设置为正,IOU小于0.3的设置为负;
4、利用上述256个anchor对应的proposal来计算loss,训练RPN;
5、然后RPN的结果还需要供给给后面的FAST-RCNN用;先从20000个proposal中,根据置信度阈值选择最高的12000个;
6、同时通过12000个进行NMS,将数量降低到2000个;
7、最后计算这2000个与GT框的IOU,大于0.5的为正,小于0.5的为负,正选32,负选96,一共128个,用于FAST-RCNN训练;
所以这里说的IOU阈值,指的是第7步的这里的IOU阈值。