faster-RCNN框架之rpn 较小目标检测,如果只使用rpn,并减少多个候选框

  通常faster-rcnn目标检测有两个步骤,一个是侯选框生成,一个是侯选框微调+目标区分,但是对于单目标识别,

我经常喜欢只使用rpn网络,效果还不错,不过仅仅的rpn使用参考的参数通常会造成一个目标很多个候选框,这时候

降低第一步骤的iou值就可以啦

first_stage_nms_iou_threshold = 0.7 (默认)

first_stage_nms_iou_threshold = 0.3