SIoU Loss

1、论文

题目:《SIoU Loss: More Powerful Learning for Bounding Box Regression》
参考博客: https://blog.csdn.net/qq_56749449/article/details/125753992

2、原理

有关IoU损失函数,像GIoU、DIoU、CIoU都没有考虑真实框与预测框之间的方向,导致收敛速度较慢,对此SIoU引入真实框和预测框之间的向量角度,重新定义相关损失函数。
SIoU loss包括四个部分:Angle cost、Distance cost、Shape cost、IoU loss

2.1 角度损失Angle cost

SIoU Loss_第1张图片
角度损失函数:
在这里插入图片描述
SIoU Loss_第2张图片
当α=π/2 或 0(即x = 1 或 0)时,角度损失最小
注意:在训练过程中,若α<π/4 ,则最小化α,否则最小化β。

2.2 距离损失Distance cost

SIoU Loss_第3张图片
距离损失函数:
在这里插入图片描述
当α→ 0时,距离损失降低;α→ π/4 时,距离损失达到最大。

2.3 形状损失Shape cost

SIoU Loss_第4张图片
形状损失函数:
在这里插入图片描述

2.4 IoU

在这里插入图片描述
IoU损失函数:
在这里插入图片描述
综上,最终SIoU loss损失函数为:
在这里插入图片描述

你可能感兴趣的:(深度学习,人工智能,计算机视觉)