bouding box回归损失函数

154 阅读1分钟
损失简述公式
IoU检测框交并比IoU=Intersection(A,B)Union(A,B)IoU = \frac{Intersection(A,B)}{Union(A,B)}
Iou loss=ln(Iou)Iou\ loss = -ln(Iou)
GIoU引入距离信息
解决boxes不重叠时,IoU失效的问题
GIoU=IoUAcuAc; 1GIoU1GIoU = IoU - \frac{A^c-u}{A^c}; \ -1\leq GIoU \leq 1
Ac是两个boxes最小外接矩形的面积,Au是面积的并集A^c是两个boxes最小外接矩形的面积,A^u是面积的并集
GIoU loss=1GIoUGIoU \ loss = 1-GIoU
DIoU引入中心点距离
区别对待不同的重合位置
DIoU=IoUd2c2; 1DIoU1DIoU = IoU - \frac{d^2}{c^2}; \ -1\leq DIoU \leq 1
d表示中心点之间的欧氏距离,c表示最小外接矩形的对角线长度d表示中心点之间的欧氏距离,c表示最小外接矩形的对角线长度
CIoUCIoU就是在DIoU的基础上增加了检测框尺度的lossCIoU=IoU(d2c2+αν)CIoU = IoU - (\frac{d^2}{c^2} + \alpha \nu)
其中αν分别表示检测框高度的比例,宽度的比例 其中\alpha 和\nu分别表示检测框高度的比例,宽度的比例