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

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

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

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

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

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

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

相关推荐
youcans_2 分钟前
【微软:多模态基础模型】(5)多模态大模型:通过LLM训练
人工智能·计算机视觉·大模型·大语言模型·多模态
飞凌嵌入式5 分钟前
飞凌嵌入式T113-i开发板RISC-V核的实时应用方案
人工智能·嵌入式硬件·嵌入式·risc-v·飞凌嵌入式
sinovoip7 分钟前
Banana Pi BPI-CanMV-K230D-Zero 采用嘉楠科技 K230D RISC-V芯片设计
人工智能·科技·物联网·开源·risc-v
搏博29 分钟前
神经网络问题之一:梯度消失(Vanishing Gradient)
人工智能·机器学习
z千鑫29 分钟前
【人工智能】深入理解PyTorch:从0开始完整教程!全文注解
人工智能·pytorch·python·gpt·深度学习·ai编程
YRr YRr38 分钟前
深度学习:神经网络的搭建
人工智能·深度学习·神经网络
威桑40 分钟前
CMake + mingw + opencv
人工智能·opencv·计算机视觉
爱喝热水的呀哈喽43 分钟前
torch张量与函数表达式写法
人工智能·pytorch·深度学习
肥猪猪爸1 小时前
使用卡尔曼滤波器估计pybullet中的机器人位置
数据结构·人工智能·python·算法·机器人·卡尔曼滤波·pybullet
LZXCyrus2 小时前
【杂记】vLLM如何指定GPU单卡/多卡离线推理
人工智能·经验分享·python·深度学习·语言模型·llm·vllm