损失函数篇 | YOLOv5 引入Unified-IoU 高质量目标检测IoU损失
论文地址:https://arxiv.org/pdf/2408.06636
开源代码地址:https://github.com/lxj-drifter/UIOU_files
目标检测是计算机视觉领域的重要组成部分,其效果直接由预测框的回归精度决定。作为模型训练的关键,IoU(交并比)很好地展示了当前预测框与真实框(Ground Truth)之间的差异。后续的研究者不断为 IoU 增加更多的考量因素,如中心距离、长宽比等。然而,仅仅细化几何差异是有上限的;此外,这些新的考量指标与 IoU 本身可能存在潜在的关联,简单地加减这些指标可能会导致“过度考虑”的问题。基于此,我