SIoU介绍

  • A+
所属分类:专业基础

SIoU loss

传统的目标检测损失函数依赖于边界框回归指标的聚合,例如预测框和真实框(即 GIoU、CIoU、ICIoU 等)的距离、重叠区域和纵横比。然而,迄今为止提出和使用的方法都没有考虑期望的真实框和预测框之间不匹配的方向。这种不足导致收敛速度较慢且效率较低,因为预测框在训练过程中可能会“四处游荡”,最终会产生一个更差的模型。

SIoU 提出了一种新的损失函数,重新定义了惩罚度量,考虑了期望回归之间的向量夹角。

SIoU =Distance cost(angle + distance)+Shape cost+IOU loss

考虑到了如下几个方面

SIoU介绍

  1. Angle costSIoU介绍
  2. Distance costSIoU介绍考虑到上面定义的Angle cost,重新定义了Distance cost
  3. shape cost定义

SIoU介绍

将 

1
SIoU 

应用于 

1
COCO-train/COCO-val

 与其他损失函数相比,提高了 

1
+2.4%

 (mAP@0.5:0.95) 和 +3.6%(mAP@0.5)

weinxin
XCJinggai的微信
这是我的微信二维码,有任何问题请加微信联系我!或发送邮件至lqzhangfengxi@163.com
zfy

发表评论

:?: :razz: :sad: :evil: :!: :smile: :oops: :grin: :eek: :shock: :???: :cool: :lol: :mad: :twisted: :roll: :wink: :idea: :arrow: :neutral: :cry: :mrgreen: