NMS算法

“”"
NMS算法步骤:
根据候选框的类别分类概率做小到大大排序:A<B<C<D<E<F

  1. 先标记最大概率矩形框F是我们需要保留下来的
  2. 从最大概率矩形框F开始,分布判断A~E与F的重叠度iou(两框的交并比)是否大于某个设定的阈值,假设B、D与F的重叠度超过阈值,那么就扔掉B、D;
  3. 从剩余的矩形框A、C、E中,选择概率最大的E, 标记为要保留下来的,然后判读E与A、C的重叠度,扔掉重叠度超过设定阈值的矩形框
    就这样一直while循环下去,直至剩下的矩形框没有了,标记完所有要保留下来的矩形框
    “”"

import numpy as np
“”"
Function: NMS algorithm
功 能:NMS算法(非极大值抑制算法)
input parameter: input_data + threshold value
输入参数:矩形框坐标数据 + 阈值
output value: the index of the datas which need to be remained
输出参数:输出需要保留的矩形框
部分资料来源于:https://blog.csdn.net/Blateyang/article/details/79113030
“”"
def py_nms(input_data, threshold):
x1 = input_data[:, 0]
y1 = input_data[:, 1

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值