NMS(non maximum suppression),中文名非极大值抑制,主要是应用在计算机视觉中的目标检测。
由于一张图片中的待检测目标,不止一个或者靠的很近,很有必要使用NMS来将目标准确提取出来。
该应用的前提是待测目标已经被框选出来候选并已经知道置信度。如下图所示。
第一步,我们看到四个候选框,置信度分别为0.95,0.83 ,0.92,0.89
我们按照从小到大顺序排列。选择第一个框,也就是下图的红框作为候选框。
第二步,找到第二个框置信度为0.92的框,然后计算他与第一个框的重叠面积(IOU)是否一个阈值(不一定为0),如果大于该阈值,则认为两个框 框的是同一个检测目标,就将其剔除队列。很明显,该框和第一个框没有重叠面积,所以保留。依次做下去,发现0.83的框与第一个大面积重合,他们是同一个目标,所以剔除,依次类推。
第三步,如下图所示,再选择第二框继续第二步操作。
最后我们得到0.95和0.92的框。
补充说明,如下图 关于IOU,可能两个框框住了两个检测目标,但是这两个检测目标太近,导致这两个框有可能重合,所以需要合理设置IOU的阈值