NMS总结

导言:

    Non-Maximum Suppression(NMS)非极大值抑制,也有个别地方称之为非最大值抑制。个人认为前者更为贴切,因为其具体实现原理是找到所有局部最大值,并抑制非局部最大值,而不是找全局最大值,后文会介绍代码。从字面意思理解,抑制那些非极大值的元素,保留极大值元素。其主要用于目标检测,目标跟踪,3D重建,数据挖掘等。

    目前NMS常用的有标准NMS, Soft  NMS, DIOU NMS等。后续出现了新的Softer NMS,Weighted NMS等改进版。

 

标准NMS

    标准NMS(左图1维,右图2维)算法伪代码如下:

NMS总结_第1张图片

NMS总结_第2张图片

左边是只计算邻域范围为3的算法伪代码。

 

实际应用

NMS总结_第3张图片

    以目标检测为例,目标检测推理过程中会产生很多检测框(A,B,C,D,E,F等),其中很多检测框都是检测同一个目标,但最终每个目标只需要一个检

你可能感兴趣的:(CV技术总结,计算机视觉,深度学习,python)