中文

FPDIoU Loss:用于旋转目标检测的高效边界框回归损失函数

计算机视觉与模式识别 2024-05-21 v2

摘要

边界框回归是目标检测的重要步骤之一。然而,旋转检测器常采用基于 SkewIoU 的更复杂损失函数,这不利于基于梯度的训练。现有旋转目标检测损失函数多仅关注两个边界框之间偏差,仅计算面积偏差或各点距离(如 LSmooth1\mathcal{L}_{Smooth-\ell 1}LRotatedIoU\mathcal{L}_{RotatedIoU}LPIoU\mathcal{L}_{PIoU})。部分损失函数计算过程极为复杂(如 LKFIoU\mathcal{L}_{KFIoU})。为提高旋转目标检测边界框回归的效率与精度,本文提出一种基于最小点距离的任意形状比较新型度量,综合考虑了现有损失函数中常见的重叠或非重叠面积、中心点距离及旋转角度等因素。我们还提出基于四点距离的损失函数 LFPDIoU\mathcal{L}_{FPDIoU},用于聚焦快速高质量锚框的精确边界框回归。实验表明,FPDIoU 损失已应用于 RTMDET、H2RBox 等前沿旋转目标检测模型训练,分别基于 DOTA、DIOR、HRSC2016 和 ICDAR 2017 RRC-MLT、ICDAR 2019 RRC-MLT 等三个旋转目标检测基准和两个任意方向场景文本检测基准,性能显著优于现有损失函数。

关键词

引用

@article{arxiv.2405.09942,
  title  = {FPDIoU Loss: A Loss Function for Efficient Bounding Box Regression of Rotated Object Detection},
  author = {Siliang Ma and Yong Xu},
  journal= {arXiv preprint arXiv:2405.09942},
  year   = {2024}
}

备注

arXiv admin note: text overlap with arXiv:2307.07662, text overlap with arXiv:1902.09630 by other authors