现在在物体检测领域,通常会使用IoU(Intersection over Union)来作为一个很重要的度量指标。但是很多时候在训练网络时,采用的是l2-norm来回归检测框的,文章认为采用l2-norm来训练模型而使用IoU来衡量模型好坏,这两者之间似乎存在着某种不匹配的关系。
举个例子,如下图所示,
现假设采用左上角右下角的坐标来表示一个矩形框,即。在采用l2-norm来度量两个框的匹配度,即上图中绿色表示的ground truth框与黑框表示的预测框之间的匹配度,当假设两个框的一个角的欧拉距离是固定的(这里假设两个框的左下角距离固定),那么这两个框在l2-norm度量下的相似度,在当以绿色框右上角为圆形某固定值为半径的圆(黑色虚线表示)上的每个点表示预测框的右上角时,l2-norm值是一样的。
这句话可以用下面公式表示:
绿色框左上角坐标为右下角为
黑色框左上角坐标为右下角为
下面要满足两个条件,
- 两个框左下角
,
距离为定值,可以用下面公式表示
这里a表示定值 - 以
为圆形,某个定值为半径,可以用下面公式表示
这里用r表示半径
将上面两式相加就是两个矩形框的l2-norm的表示,可以看出只要取圆上的另一一点,l2-norm的结果是一样的,而这时候IoU和后面要介绍的GIoU值都是不一样的,这就说明了l2-norm不能很好的表示IoU值。而且这个结论与用什么方式表示矩形框是无关的,如下面用矩形框的中心点和框的宽高来表示一个框也可以得到相应的结论。
为了更好的优化网络以及有一个更好的度量,文章提出了GIoU的概念。
一、GIoU的性质
下面先来回顾一下IoU的计算,如下式表示
对于任意两个形状,A和B,它们的IoU为
当两个物体没有相交时,IoU定义为0。
IoU具有以下性质,使得IoU在2D、3D计算机视觉中成为一个重要的评测指标:
- IoU能作为一中度量,而
能满足度量的所有性质。
- IoU对于尺度有不变性。这意味着两个任意形状A和B之间的相似性与它们的空间尺度无关。
然而IoU作为度量也有它的缺点,那就是当=0是IoU为0,这时候并不能表示两个物体之间的距离,他们是邻近的还是相距很远的。为了解决这个问题,本文提出GIoU。
先介绍一下GIoU作为一种度量方式,它具有的性质:
- 类似与IoU,GIoU也能表示距离。
能满足度量的所有性质。
- 类似IoU,GIoU也对于尺度有不变性。
- GIoU是IoU的下界,即
,当A和B的形状和位置无限趋近时,等号成立
- 类似
,
,如果两个物体重叠时
,如果两个物体的并集与包围两个物体的形状之比趋近0时GIoU取得-1。
二、GIoU的计算方式
对于两个框表示
预测框:
标注框:
- 为确保预测框中
,
:
- 标注框的面积为
- 预测框的面积为
- 两者的交集为
- 计算两个框最小的包围框
- 最小的包围框的面积为
-
当IoU为0时在训练网络时没有梯度了,影响训练质量和模型收敛速度,而GIoU不会为0。而且从GIoU的性质3可以看出,GIoU与IoU有很强的相关性。文章也采用了随机的10K个样本求IoU和GIoU得到下图,也可以 看出二者相关性,并且还能看出GIoU相对于IoU能够获得更陡的梯度值,更有利于模型训练。
3.png
关于GIoU Loss的稳定性
通过前面介绍知道
,
所以两个loss都是有界的
在IoU为0时的行为
当IoU=0时,,要使得loss减小,就要使
增大,从而要使得
增大或者
减小,要达到这种要求就需要预测框和标注框尽量重叠。
关于GIoU实践可能存在的问题
在博主训练yolov3的时候,使用了GIoUloss,不用与smoothL2-loss的计算只要算坐标的一个偏置量的差值,要算GIoUloss必须先求出坐标,而在求坐标的时候使用了e为底的指数,造成了w和h为inf的情况,导致loss为nan,采用的办法是限制w和h为一个较大的值。