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

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

2.2 距离损失Distance cost

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

2.3 形状损失Shape cost

在这里插入图片描述
形状损失函数:
在这里插入图片描述

2.4 IoU

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

猜你喜欢

转载自blog.csdn.net/Chenjiahui_LYee/article/details/128603242