计算机视觉中的NMS非极大值抑制

NMS 是"非极大抑制"(Non-Maximum Suppression)的缩写,是一种在目标检测算法中广泛使用的技术。它的主要目的是减少目标检测过程中的多余的边界框,以便只保留最佳的一个边界框。

在目标检测任务中,算法会对图像中可能出现目标的区域进行评分,并为它们生成边界框。由于多个边界框可能覆盖到同一个目标,因此需要一个方法来选择最合适的边界框。NMS 通过以下步骤实现这一目的:

  1. 选择置信度最高的边界框(即目标存在的可能性最高的边界框)。

  2. 抑制与这个边界框高度重叠的所有其他边界框。具体来说,会计算这些边界框与置信度最高的边界框的交并比(IoU),如果IoU超过某个阈值(例如0.5),则这些边界框会被认为是对同一个目标的多余检测,并因此被抑制。

  3. 从剩下的边界框中再次选择置信度最高的边界框,重复上述过程,直到所有的边界框都被处理完毕。

通过NMS,我们可以确保每个目标只被检测一次,从而提高目标检测的准确性。这一技术在各种目标检测模型如YOLO、SSD、Faster R-CNN等中都有应用。

相关推荐
武子康10 分钟前
h3.c 的两个权重槽,怎样把内存压力换成 SSD 等待
c语言·人工智能·agent
大模型真好玩13 分钟前
豆包真人分身操作技巧:这次让我来做主演
人工智能·agent·deepseek
xyzzklk26 分钟前
在Open-WebUI中提升向量化速度
网络·人工智能·爬虫·ai·语言模型·自然语言处理·安卓
QXWZ_IA27 分钟前
矿山安全监测从感知到预警的全链路算法
人工智能·安全
日日大王学编程1 小时前
AI Agent 技术发展历程、演进逻辑与技术范式综述
人工智能·python
Yuzhiyuxia1 小时前
【AI Agent面试系列】上下文管理1
人工智能·面试·职场和发展
王中阳Go2 小时前
预算给够了,时间花错了地方:三跳链路给了 27 秒,实际只用掉 9 秒
人工智能·后端·性能优化
方方洛2 小时前
ray教程-25-附录E-端到端实战案例
人工智能·分布式·机器学习
火山引擎开发者社区2 小时前
上万个 Agent 在线不乱跑,云沙箱给 Agent 造了间“独立办公室”
人工智能
神鹰手哥_起飞版2 小时前
一句话生成带品牌 logo 的架构图:方舟 Seed 2.1 负责听懂,代码负责画准
人工智能