当前位置：首页 > news >正文

【目标检测】非极大值抑制(NMS)的原理与实现

news 来源：原创 2025/6/20 14:53:36

🧑 博主简介：曾任某智慧城市类企业算法总监，目前在美国市场的物流公司从事高级算法工程师一职，深耕人工智能领域，精通python数据挖掘、可视化、机器学习等，发表过AI相关的专利并多次在AI类比赛中获奖。CSDN人工智能领域的优质创作者，提供AI相关的技术咨询、项目开发和个性化解决方案等服务，如有需要请站内私信或者联系任意文章底部的的VX名片（ID：xf982831907）

💬 博主粉丝群介绍：① 群内初中生、高中生、本科生、研究生、博士生遍布，可互相学习，交流困惑。② 热榜top10的常客也在群里，也有数不清的万粉大佬，可以交流写作技巧，上榜经验，涨粉秘籍。③ 群内也有职场精英，大厂大佬，可交流技术、面试、找工作的经验。④ 进群免费赠送写作秘籍一份，助你由写作小白晋升为创作大佬。⑤ 进群赠送CSDN评论防封脚本，送真活跃粉丝，助你提升文章热度。有兴趣的加文末联系方式，备注自己的CSDN昵称，拉你进群，互相学习共同进步。

在这里插入图片描述

@TOC

一、引言

在目标检测中，NMS是提升模型性能的关键后处理技术，但90%的初学者对其实现原理理解不足。本文将彻底解密NMS的工作原理，并提供可直接运行的Python代码，让你真正掌握这一核心技能！

二、为什么需要非极大值抑制(NMS)？

目标检测模型在处理图像时，往往会对同一个目标产生多个重叠的预测框。如下图所示：

图片来源于：https://blog.csdn.net/weixin_62264287/article/details/133936328

这些重叠框会导致：

重复检测：同一目标被多次识别
结果冗余：输出包含大量无效预测
性能下降：影响后续处理效率

非极大值抑制(Non-Maximum Suppression, NMS) 正是为了解决这一问题而诞生。它像一位"裁判"，从众多候选框中选出最合适的一个，淘汰冗余预测。

三、NMS核心原理揭秘

3.1 NMS工作原理四步曲

排序候选框：将所有预测框按置信度(confidence score)从高到低排序
选择最高分：选取置信度最高的框作为保留框
计算IOU：计算该保留框与剩余所有框的IOU
剔除重叠框：删除IOU超过阈值的框（通常阈值设为0.5）

重复步骤2-4，直到所有框都被处理完毕。整个过程如下图所示：

四、Python实现NMS算法

下面我们实现一个完整的NMS算法，并可视化处理效果：

import numpy as np
import cv2
import matplotlib.pyplot as plt
import matplotlib.patches as patchesdef nms(boxes, scores, threshold=0.5):"""非极大值抑制(NMS)实现参数:boxes: 边界框列表 [x1, y1, x2, y2]scores: 每个框的置信度threshold: IOU阈值返回:保留的框索引列表"""# 如果没有框，直接返回空列表if len(boxes) == 0:return []# 将框坐标转换为float类型boxes = np.array(boxes, dtype=np.float32)# 获取框的坐标x1 = boxes[:, 0]y1 = boxes[:, 1]x2 = boxes[:, 2]y2 = boxes[:, 3]# 计算每个框的面积areas = (x2 - x1 + 1) * (y2 - y1 + 1)# 按置信度降序排序idxs = np.argsort(scores)# 初始化保留列表keep = []while len(idxs) > 0:# 取出当前置信度最高的框last = len(idxs) - 1i = idxs[last]keep.append(i)# 计算当前框与其他框的IOUxx1 = np.maximum(x1[i], x1[idxs[:last]])yy1 = np.maximum(y1[i], y1[idxs[:last]])xx2 = np.minimum(x2[i], x2[idxs[:last]])yy2 = np.minimum(y2[i], y2[idxs[:last]])# 计算交集区域的宽和高w = np.maximum(0, xx2 - xx1 + 1)h = np.maximum(0, yy2 - yy1 + 1)# 计算IOUintersection = w * hiou = intersection / (areas[i] + areas[idxs[:last]] - intersection)# 删除IOU大于阈值的框delete_idxs = np.where(iou > threshold)[0]idxs = np.delete(idxs, np.concatenate(([last], delete_idxs)))return keepdef visualize_boxes(image, boxes, scores, title, color='r'):"""可视化边界框"""plt.figure(figsize=(10, 6))plt.imshow(image)ax = plt.gca()for i, box in enumerate(boxes):x1, y1, x2, y2 = boxrect = patches.Rectangle((x1, y1), x2-x1, y2-y1, linewidth=2, edgecolor=color, facecolor='none')ax.add_patch(rect)plt.text(x1, y1-10, f'{scores[i]:.2f}', fontsize=10, color='white', bbox=dict(facecolor=color, alpha=0.8))plt.title(title)plt.axis('off')plt.show()# 创建模拟图像
image = np.ones((400, 600, 3), dtype=np.uint8) * 255
cv2.putText(image, 'Target Detection', (150, 200), cv2.FONT_HERSHEY_SIMPLEX, 2, (0, 0, 0), 3)# 模拟检测结果 (格式: [x1, y1, x2, y2])
boxes = [[100, 100, 300, 300],   # 高置信度[120, 120, 320, 320],   # 与第一个高度重叠[150, 150, 350, 350],   # 与第一个高度重叠[400, 150, 550, 350],   # 不同位置[380, 130, 530, 330],   # 与第四个重叠[420, 170, 570, 370]    # 与第四个重叠
]# 对应的置信度
scores = [0.95, 0.90, 0.85, 0.92, 0.88, 0.80]# 可视化原始检测结果
visualize_boxes(image, boxes, scores, '原始检测结果 (NMS前)', color='r')# 应用NMS
keep = nms(boxes, scores, threshold=0.5)# 筛选保留的框
nms_boxes = [boxes[i] for i in keep]
nms_scores = [scores[i] for i in keep]# 可视化NMS后的结果
visualize_boxes(image, nms_boxes, nms_scores, 'NMS处理后的结果', color='g')

五、代码解析与运行结果

5.1 核心函数解析

nms函数：
- 输入：边界框列表、置信度列表、IOU阈值
- 处理流程：
  - 按置信度排序
  - 循环处理每个框
  - 计算当前框与其他框的IOU
  - 删除IOU超过阈值的框
- 输出：保留框的索引列表
visualize_boxes函数：
- 功能：在图像上绘制边界框和置信度
- 参数：图像、框列表、置信度列表、标题、颜色

5.2 运行效果展示

运行上述代码，你将看到两个对比图像：

NMS前：

红色框表示原始检测结果
同一目标有多个重叠框

NMS后：

绿色框表示NMS处理后的结果

六、NMS的进阶应用与优化

6.1 软性NMS(Soft-NMS)

传统NMS直接删除重叠框，可能导致以下问题：

密集目标漏检
阈值设置敏感

软性NMS的改进方案：

# 在NMS循环中替换删除操作
for j in delete_idxs:# 根据IOU降低置信度，而不是直接删除scores[j] = scores[j] * (1 - iou[j])

6.2 多类别NMS处理

当处理多类别检测时，需要：

# 按类别分组处理
for class_id in unique_classes:class_indices = np.where(class_ids == class_id)[0]class_boxes = boxes[class_indices]class_scores = scores[class_indices]keep_indices = nms(class_boxes, class_scores)# 合并各类别结果

6.3 其他改进方法

方法	核心思想	适用场景
Weighted NMS	加权融合重叠框	需要更精确位置
Adaptive NMS	动态调整阈值	密集目标检测
DIoU NMS	考虑中心点距离	解决边界框错位

七、NMS在实际项目中的应用技巧

7.1 阈值选择策略

通用目标：0.5-0.7
密集小目标：0.3-0.5
大目标/精确检测：0.7-0.9

7.2 性能优化技巧

# 向量化加速
def vectorized_nms(boxes, scores, threshold):# 使用矩阵运算代替循环x1 = boxes[:, 0]y1 = boxes[:, 1]x2 = boxes[:, 2]y2 = boxes[:, 3]areas = (x2 - x1 + 1) * (y2 - y1 + 1)order = scores.argsort()[::-1]keep = []while order.size > 0:i = order[0]keep.append(i)# 向量化计算IOUxx1 = np.maximum(x1[i], x1[order[1:]])yy1 = np.maximum(y1[i], y1[order[1:]])xx2 = np.minimum(x2[i], x2[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 * hiou = inter / (areas[i] + areas[order[1:]] - inter)# 向量化索引inds = np.where(iou <= threshold)[0]order = order[inds + 1]return keep