yunyang tensorflow-yolov3 NMS:non maximum suppression 非极大值抑制方法

NMS: non maximum suppression

非极大值抑制的背景

生成器对一张图片生成的框在进行图外滤去、颠倒滤去、低概率滤去之后,会留下一些符合要求的框,但这些框有不少部分是重合的,而在同一张图片中,我们需要识别的目标是有限的,所以我们要对这些框继续筛选并剔除,以满足我们目标与框一一对应的关系。
yunyang tensorflow-yolov3 NMS:non maximum suppression 非极大值抑制方法

非极大值抑制的步骤

在每个框给出得分的前提下,我们按照以下步骤实现非极大值抑制方法:

  1. 统计图片中所有识别出目标的类,对每个类单独分析
  2. 遍历图片中某个类的所有的框,找出得分最高的框a(图中红色标注)
    yunyang tensorflow-yolov3 NMS:non maximum suppression 非极大值抑制方法
  3. 剔除与框a的IOU(交并比,一定程度上表示框的面积的重合率)大于一定阈值的框
    yunyang tensorflow-yolov3 NMS:non maximum suppression 非极大值抑制方法
  4. 继续遍历余下的框,找出余下的框中的分最高的框,重复步骤2,直到这个类中没有余下需处理的框
    yunyang tensorflow-yolov3 NMS:non maximum suppression 非极大值抑制方法
  5. 重复步骤1

参考文章:NMS——非极大值抑制