NMS算法实现

NMS算法(非极大值抑制)是目标检测算法中经典的后处理步骤,其本质是搜索局部最大值,抑制非极大值元素。主要利用目标检测框以及对应的置信度分数,设置一定的阈值来删除重叠较大的边界框。

其算法流程如下:

根据置信度得分进行排序

选择置信度最高的目标检测框添加到输出列表中,将其从检测框列表中删除

计算该检测框与剩余候选检测框的IOU

删除IOU大于阈值的检测框

重复上述4步,直至检测框列表为空

复制代码
import numpy as np


def nms(dets, thresh):  # x1, y1, x2, y2, score
    x1, y1, x2, y2, scores = dets[:, 0], dets[:, 1], dets[:, 2], dets[:, 3], dets[:, 4]
    areas = (x2 - x1 + 1) * (y2 - y1 + 1)  # 各个方框的面积
    order = scores.argsort()[::-1]  # 按置信度排序后的index, 作为候选集
    keep = []  # 保存筛选出来的方框的index
    while order.size > 0:

        i = order[0]  # 当前置信度最大的方框
        keep.append(i)
        xx1 = np.maximum(x1[i], x1[order[1:]])
        xx2 = np.minimum(x2[i], x2[order[1:]])
        yy1 = np.maximum(y1[i], y1[order[1:]])
        yy2 = np.minimum(y2[i], y2[order[1:]])

        w = np.maximum(0.0, (xx2 - xx1 + 1))
        h = np.maximum(0.0, (yy2 - yy1 + 1))
        inter = w * h  # 当前置信度最大的框和其他所有框的相交面积
        overlap = inter / (areas[i] + areas[order[1:]] - inter)
        inds = np.where(overlap <= thresh)[0]  # 交并比小于thresh的仍然保留在候选集里, 大的过滤掉
        order = order[inds + 1]  # inds + 1对应原来order中overlap小于thresh的项
    return keep


if __name__ == '__main__':
    detections = [
        [10, 20, 100, 100, 0.9],
        [20, 10, 110, 100, 0.88],
        [20, 20, 110, 110, 0.86],
        [40, 50, 200, 200, 0.95],
        [45, 52, 198, 202, 0.87]
    ]
    detections = np.array(detections)
    keeps = nms(detections, 0.5)
    print(detections[keeps])
相关推荐
胖咕噜的稞达鸭15 分钟前
C++中的父继子承:继承方式实现栈及同名隐藏和函数重载的本质区别, 派生类的4个默认成员函数
java·c语言·开发语言·数据结构·c++·redis·算法
笑口常开xpr22 分钟前
【C++】模板 - - - 泛型编程的魔法模具,一键生成各类代码
开发语言·数据结构·c++·算法
IT小番茄1 小时前
Kubernetes云平台管理实战:自动加载到负载均衡(七)
算法
笑口常开xpr1 小时前
【C++继承】深入浅出C++继承机制
开发语言·数据结构·c++·算法
让我们一起加油好吗2 小时前
【基础算法】DFS
算法·深度优先
爱学习的小鱼gogo4 小时前
python 矩阵中寻找就接近的目标值 (矩阵-中等)含源码(八)
开发语言·经验分享·python·算法·职场和发展·矩阵
红纸2814 小时前
Subword算法之WordPiece、Unigram与SentencePiece
人工智能·python·深度学习·神经网络·算法·机器学习·自然语言处理
CUMT_DJ4 小时前
从零复现论文(1)——通感一体化实现协作基站分配与资源分配(CBARA)策略
算法·通感一体化
tt5555555555555 小时前
CSDN 教程:C++ 经典字符串与栈算法题逐行详解
c++·算法·哈希算法
_dindong5 小时前
基础算法:滑动窗口
数据结构·学习·算法·leetcode·力扣