码迷,mamicode.com
首页 > 其他好文 > 详细

目标检测 — NMS

时间:2018-09-06 17:59:42      阅读:226      评论:0      收藏:0      [点我收藏+]

标签:目标检测   大于   多个   size   算法   预测   没有   bsp   效果   

1、非极大值抑制步骤

  非极大值抑制算法(Non-maximum suppression,NMS)在目标检测中经常用到。我们的检测算法可能对同一目标产生多次检测的结果,非极大值抑制算法可以保证每个目标只检测一次,找到检测效果最好的框。

(1)去除所有预测框置信度于某个阈值的框,这里的阈值选取为0.5。

(2)在剩下的1、2、4、5、6号框中选取置信度最大的,即图中5号框,比较除5号框以外所有框于5号框的交并比,若交并比大于某个阈值(人工给定的值),则抑制这些不是局部极大值预测框输出。

(3)再从剩下的框中选取置信度最大的,重复上面的步骤,直至没有框被舍去时结束。

(4)最后2号和5号框被保留,其他框被抑制。

  若检测物体是多个类别的时候,一般情况是对每个类别分别做一次非极大值抑制算法。

 

目标检测 — NMS

标签:目标检测   大于   多个   size   算法   预测   没有   bsp   效果   

原文地址:https://www.cnblogs.com/eilearn/p/9599269.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!