opencv 非极大值抑制
时间: 2023-09-15 16:20:53 浏览: 175
非极大值抑制C++/C/Matlab版的源码
非极大值抑制(Non-Maximum Suppression,简称NMS)是一种常用的目标检测算法,也可以应用于角点检测、边缘检测等领域。
在OpenCV中,可以通过cv2.dnn.NMSBoxes()函数实现非极大值抑制。该函数需要传入一个矩形框列表、置信度得分列表、置信度阈值、IoU(交并比)阈值等参数,返回经过NMS处理后保留的矩形框的索引。
具体实现过程如下:
1. 首先,将矩形框按照置信度得分进行降序排序。
2. 选取得分最高的矩形框,并将其加入最终结果列表中。
3. 遍历剩余的矩形框,如果与已选中的矩形框的IoU值大于等于设定的阈值,则将其从候选列表中删除。
4. 重复步骤2和3,直到没有更多的矩形框需要处理。
通过上述步骤,可以得到经过非极大值抑制处理后的矩形框列表,从而完成目标检测任务。
阅读全文