深圳Yolo 视觉
深圳
立即加入
词汇表

联盟路交叉口IoU)

了解什么是 "交集大于联合IoU)、它是如何计算的,以及它在物体检测和人工智能模型评估中的关键作用。

联合相交IoU) 是计算机视觉IoUCV) 中使用的一个基本评估指标。 计算机视觉 (CV)中用来衡量物体检测器在特定数据集上的准确性。 物体检测器在特定数据集上的准确性。在学术界通常被称为 IoU 量化了两个区域之间的重叠程度。 两个区域之间重叠程度。 模型生成的预测边界框与地面实况边界框之间的重叠程度,地面实况边界框是手工标注的注释,代表了物体的实际位置。 对象的实际位置。这个度量产生的值从 0 到 1 不等,0 表示没有重叠,1 表示预测与实际完全匹配。 预测与实际情况完全吻合。

了解IoU 如何工作

IoU 的计算在概念上简单明了,因此成为IoU 检测的标准基准。 物体检测图像分割任务的标准基准。计算方法是 将交集区域(重叠区域)除以结合区域(两个方框的总合区域 方框的总和)。通过将重叠区域与总面积进行归一化,IoU 提供了一种尺度不变的测量方法,即 即无论对象的大小如何,它都能评估拟合的质量。

在实际应用中,通常会对IoU 分数设置一个阈值,以便classify 预测分为 "正 "或 "负 "检测。 正 "或 "负 "检测。例如,在 COCO 数据集基准等标准挑战中 COCO 数据集基准等标准挑战中,通常使用 0.5(或 50%)通常用于确定检测是否有效。这个阈值对于计算下游的 精度召回率等下游指标至关重要,这些指标最终会被纳入 平均精度 (mAP)分数,用于对模型性能进行排名 用于对模型性能进行排名。

用Python计算IoU

您可以使用以下工具中提供的效用函数,轻松计算出盒子之间的IoU ultralytics 包。这对自定义评估脚本或在推理过程中理解模型行为非常有用。 推理过程中的模型行为。

import torch
from ultralytics.utils.metrics import box_iou

# Define a ground truth box and a predicted box in (x1, y1, x2, y2) format
# Example: A 100x100 pixel box starting at (50,50) vs shifted prediction
ground_truth = torch.tensor([[50, 50, 150, 150]])
prediction = torch.tensor([[60, 60, 160, 160]])

# Calculate the Intersection over Union
iou_score = box_iou(ground_truth, prediction)

print(f"IoU Score: {iou_score.item():.4f}")
# Output represents the overlap ratio

实际应用

IoU 所提供的精确定位功能对各行各业都至关重要,因为在这些行业中,安全和精确度是 至关重要。

  • 人工智能在汽车领域的应用在 在开发自动驾驶汽车时,系统 系统必须准确区分行人、其他车辆和静态障碍物。高IoU 分数可确保 感知系统准确地知道物体相对于汽车的位置,而不仅仅是知道它的存在。这种精确的 这种精确定位对于轨迹规划和防碰撞系统至关重要,Waymo 和特斯拉等领先企业都在使用这种系统。 Waymo和特斯拉等领先企业所使用的轨迹规划和避免碰撞系统。
  • 医学图像分析 当使用人工智能detect 核磁共振成像或 CT 扫描中的肿瘤等异常情况时,病变的确切边界对诊断和治疗规划非常重要。 对诊断和治疗规划至关重要。具有高IoU 的模型可确保 医疗工具中的人工智能能够准确地 突出显示整个受影响区域,而不包括过多的健康组织,从而协助放射科医生做出关键决策。 关键决策。

模型训练和推理中的IoU

除了简单的评估,IoU 还在现代架构的内部机制中发挥着积极作用,例如 YOLO11.

  • 损失函数优化: 传统的损失函数往往难以处理不重叠的边界框。为了解决这个问题,我们采用了各种变体,如 广义IoU GIoU)、距离IoU DIoU) 和完整IoU CIoU) 等变体被直接纳入训练过程。 过程中。即使在方框不接触的情况下,这些变化也能提供梯度,从而帮助模型更快地收敛,实现更好的定位。 实现更好的定位。
  • 非最大抑制(NMS) 在推理过程中,物体检测器可能会为一个物体输出多个边界框。NMS 使用IoU 过滤这些结果。如果两个预测框的IoU 值高于特定NMS 阈值(如 0.7),且属于同一类别,则IoU 值较低的那个预测框将被排除。 属于同一类别,则置信度较低的那个 置信度分数较低的预测框会被删除,只留下最佳预测框。 预测。

IoU 与相关指标的比较

必须将IoU 与类似术语区分开来,以便为特定任务选择正确的衡量标准。

  • IoU 与精度:准确性通常 通常衡量正确分类的百分比(例如,"这是一只猫吗?)它不考虑 物体在哪里。IoU 专门测量定位的质量(例如,"盒子与猫的匹配程度如何? 盒子与猫的匹配程度如何?")。如果盒子松动或错位,模型的分类准确率可能很高,但IoU 却很低。 错位。
  • IoU 与骰子系数 虽然两者都测量重叠度并呈正相关,但 Dice 系数(或像素的 F1 分数)更看重交叉点。 交叉的权重。在语义分割任务中,Dice 分数通常被优先考虑。 而IoU 则是边界框对象检测标准。 则是边界框对象检测的标准。

加入Ultralytics 社区

加入人工智能的未来。与全球创新者联系、协作和共同成长

立即加入