中文

Shape-IoU:考虑边界框形状和尺度的更准确度量

计算机视觉与模式识别 2024-01-15 v2

摘要

作为检测器定位分支的重要组成部分,边界框回归损失在目标检测任务中起着重要作用。现有的边界框回归方法通常考虑GT框与预测框之间的几何关系,并利用边界框的相对位置和形状计算损失,而忽略了边界框自身属性(如形状和尺度)对边界框回归的影响。为弥补现有研究的不足,本文提出了一种关注边界框自身形状和尺度的边界框回归方法。首先,我们分析了边界框的回归特性,发现边界框自身的形状和尺度因子会对回归结果产生影响。基于上述结论,我们提出了Shape IoU方法,该方法通过关注边界框自身的形状和尺度来计算损失,从而使边界框回归更加准确。最后,我们通过大量对比实验验证了我们的方法,结果表明我们的方法能有效提升检测性能,并优于现有方法,在不同检测任务中达到了最先进的性能。代码可在https://github.com/malagoutou/Shape-IoU获取。

关键词

引用

@article{arxiv.2312.17663,
  title  = {Shape-IoU: More Accurate Metric considering Bounding Box Shape and Scale},
  author = {Hao Zhang and Shuaijie Zhang},
  journal= {arXiv preprint arXiv:2312.17663},
  year   = {2024}
}