非极大值抑制(NMS)(Non-Maximum Suppression,NMS)机器改进(soft nms, learning nms)待更新2019-9-29

      非极大值抑制(NMS)顾名思义就是抑制舍弃掉得分不高的元素,搜索局部的极大值。此处不讨论通用的NMS算法,仅关注用于目标检测中用于提取分数最高的窗口的NMS。例如在行人检测中,滑动窗口经提取特征,经分类器分类识别后,每个窗口都会得到一个分数。但是滑动窗口会导致很多窗口与其他窗口存在包含或者大部分交叉的情况。这时就需要用到NMS来选取那些是行人的概率最大,并且抑制舍弃掉那些分数低的窗口。

 

RCNN或其他的神经网络模型会从一张图片中找出n个可能是物体的矩形框,然后为每个矩形框为做类别分类概率,就像上面的图片一样,为定位一个车辆,最后算法找出了一堆的方框,我们需要判别哪些矩形框是没用的。我们还是从一个例子中来阐述下nms是如何做的.

假设网络预测出了6个矩形框A1~A6,对每个box按照置信度从小到大做排序,排队结果为:A1,A2,A3,A4,A5,A6
(1)从最大置信度矩形框A6开始,分别判断A1,A2,A3,A4,A5与A6的重叠度IOU是否大于某个设定的阈值(假设阈值a=0.7);
(2)假设A2、A3与A6的重叠度超过阈值,那么就扔掉A2、A3;并标记第一个矩形框A6,此时只剩下了矩形框A1,A4,A5,A6;
(3)从剩下的矩形框A1、A4、A5中,选择概率最大的A5,然后判断A1,A4与A5的重叠度,若重叠度大于设定的阈值,那么就扔掉这个box;假设扔掉了A1,则此时只剩下了矩形框A4,A5,A6;

(4)这样就找到所有被保留下来的

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值