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

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

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

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

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

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

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

相关推荐
胡耀超14 分钟前
知识图谱入门——3:工具分类与对比(知识建模工具:Protégé、 知识抽取工具:DeepDive、知识存储工具:Neo4j)
人工智能·知识图谱
陈苏同学22 分钟前
4. 将pycharm本地项目同步到(Linux)服务器上——深度学习·科研实践·从0到1
linux·服务器·ide·人工智能·python·深度学习·pycharm
吾名招财41 分钟前
yolov5-7.0模型DNN加载函数及参数详解(重要)
c++·人工智能·yolo·dnn
鼠鼠龙年发大财1 小时前
【鼠鼠学AI代码合集#7】概率
人工智能
龙的爹23331 小时前
论文 | Model-tuning Via Prompts Makes NLP Models Adversarially Robust
人工智能·gpt·深度学习·语言模型·自然语言处理·prompt
工业机器视觉设计和实现1 小时前
cnn突破四(生成卷积核与固定核对比)
人工智能·深度学习·cnn
我算是程序猿3 小时前
用AI做电子萌宠,快速涨粉变现
人工智能·stable diffusion·aigc
萱仔学习自我记录3 小时前
微调大语言模型——超详细步骤
人工智能·深度学习·机器学习
湘大小菜鸡3 小时前
NLP进阶(一)
人工智能·自然语言处理
XiaoLiuLB3 小时前
最佳语音识别 Whisper-large-v3-turbo 上线,速度更快(本地安装 )
人工智能·whisper·语音识别