NMS(Non-maximum suppression)非极大抑制

一、NMS作用

NMS的作用:去掉detection任务重复的检测框。

简单的说,就是不是局部最大值的检测框都去掉。

可以被理解是局部最大搜索,这个局部代表的是一个邻域,邻域有两个参数可变,一是邻域的维数,二是邻域的大小(《Efficient Non-Maximum Suppression》)

NMS(Non-maximum suppression)非极大抑制_第1张图片

二、NMS的原理

RPN为每个框计算出一个score,score越大,则证明框越接近期待值。

如果目标分别有多个选择框,要去掉多余的选择框,分别在局部选出最大框,然后去掉和这个框IOU<0.7的框。

基本算法流程(抑制的过程是一个迭代-遍历-消除的过程):

  1. 将所有框的得分排序,选中最高分及其对应的框;
  2. 遍历其余的框,如果和当前最高分框的重叠面积(IOU)大于一定的阈值,将框删除;

你可能感兴趣的:(Deep,Learning,NMS)