NMS算法

一、IoU(交并比)

计算交集:

inter=|x_{12}-x_{21}|*|y_{12}-y_{21}|

计算并集:

union=|x_{11}-x_{12}|*|y_{11}-y_{12}|+|x_{21}-x_{22}|*|y_{21}-y_{22}|-inter

计算IoU:

IoU=\frac{inter}{union}

二、NMS(非极大值抑制)

以目标检测为例:

        目标检测的过程中在同一目标的位置上会产生大量的候选框,这些候选框相互之间可能会有重叠,此时我们需要利用非极大值抑制找到最佳的目标边界框,消除冗余的边界框。

例:

        左图是人脸检测的候选框结果,每个边界框有一个置信度得分(confidence score),如果不使用非极大值抑制,就会有多个候选框出现。右图是使用非极大值抑制之后的结果,符合我们人脸检测的预期结果。

非极大值抑制的流程如下:

输入:目标边界框列表及其对应的置信度得分列表,设定阈值,阈值用来删除重叠较大的边界框

①根据置信度得分进行排序

②选择置信度最高的比边界框添加到最终输出列表中,将其从边界框列表中删除

③计算置信度最高的边界框与其它候选框的IoU。

④删除IoU大于阈值的边界框

⑤重复②③④,直至边界框列表为空。

python代码实现:

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
Java NMS(Non-Maximum Suppression)算法是一种用于目标检测和边界框过滤的算法。 在目标检测中,通常会产生多个候选边界框,并希望从中选择出最佳的边界框。但是,如果边界框之间有重叠,那么仅仅基于得分来选择最佳边界框可能会导致重复检测。 NMS算法可以帮助我们解决这个问题。它的基本思想是通过抑制那些与得分更高的边界框存在重叠的边界框,从而选择最佳的边界框。算法步骤如下: 1. 对所有的边界框按照得分进行降序排序。 2. 选择排在第一个位置的边界框作为最佳边界框。 3. 计算最佳边界框与其他边界框的重叠面积,如果重叠面积超过预定义的阈值,则将此边界框舍弃。 4. 从剩下的未处理边界框中选择得分最高的边界框,并重复步骤3。 5. 重复步骤4直到所有的边界框都被处理。 经过NMS算法处理后,我们可以得到一组没有重复检测的最佳边界框。 在Java中实现NMS算法,我们可以借助Java提供的排序方法,如Collections.sort()来对边界框进行排序。然后,我们可以使用一个循环来遍历边界框并计算重叠面积,并根据阈值的设定进行舍弃操作。最后,我们可以得到最终的结果。 总的来说,Java NMS算法是一种用于边界框过滤和目标检测的有效算法,它可以帮助我们选择最佳的边界框并避免重复检测。通过对边界框的排序和重叠面积的计算,我们可以通过一定规则来筛选出所需的最佳边界框。

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值