非极大抑制

非极大值抑制(Non-Maximum Suppression,NMS)正如其名,如果不是极大值,就将这个结果抑制掉。

目标检测中,利用检测方法检测到的结果往往有很多,而我们一般只需要最好的那一个结果。

非极大抑制就是以这些检测结果(置信度)的最大的那个检测框作为基准,再对其他结果进行排序。

按顺序从排序结果中取出其他结果,如果其他检测结果和最大结果的IOU(可以暂时理解为重叠面积)大于我们规定的阈值的话,就说明这个结果是冗余的,就可以把这些结果去掉。

 

这里写图片描述

如图所示,最初的检测结果有5个,分别为:女:0.98,0.83,0.75  男:0.81,0.67

这5个框最大值是0.98那个框,将这个框加入到最终检测结果列表D中,再对其他4个检测框按照置信度又高到低进行排序存放到列表B中,分别为:0.83(女),0.81(男),0.75(女),0.67(男)#有时也会按照右下角坐标,或者检测框大小排序

以0.98这个框为基准,依次同列表B中的每个框计算IOU,如果大于某个值(假设我们规定0.5,一般情况这个阈值范围为0.3-0.5)的话就将B中的框去掉,过程如下:

  1. 0.83(女)同0.98(女)的IOU大于0.5,去掉;
  2. 0.81(男)同0.98的IOU为0,保留;
  3. 0.75(女)同0.98(女)IOU也大于0.5,去掉;
  4. 0.67(男)同0.98(女)IOU 为0,保留;

这里写图片描述

现在列表B中只剩下0.81(男)和0.67(男)两个检测结果;#上图绿色框

最终检测结果列表D中0.98(女)#上图红色框

  1. 在列表B中继续取出置信度最高的那个检测框,放入到最终检测结果D中。#此时最终检测结果中有两个框0.98那个和0.81这两个,列表B中只剩下0.67(男)
  2. 将列表B中剩下的框0.67(男)同0.81(男)对比IOU,大于0.5,去掉

这样最后结果就只剩下了0.98(女)跟0.81(男)这两个检测框,正是我们想要的检测结果。

这里写图片描述

IOU解释:

IOU=两个图形的交/两个图形的并

参考:

1、https://www.cnblogs.com/makefile/p/nms.html

2、https://blog.csdn.net/shuzfan/article/details/52711706

 

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值