定位误差常用的损失函数
前言
在目标检测任务中,为了更好的量化定位损失,各路大神想了很多办法。从最开始的针对xywh的均方误差,到IOU损失,再到后续的GIOU损失、DIOU损失、CIOU损失,越来越复杂。其中均方误差比较简单,不再介绍。重点讲一下IOU系列。
IOU损失
这里牵涉到两个概念,“IOU”和“IOU损失”,对于前者,已经很熟悉了,不再赘述。而对于后者,则是简单地令
这样,IOU越接近于1,损失越接近于0。并且相比于均方误差没有尺度大小影响wh损失的现象。但美中不足的是,当I=0,即两个框不相交时无法正确计算。
GeneralizedIOU损失
IOU越大,损失越小。D1和D2占S的比例越大,IOU越大。
DistanceIOU损失
IOU越大,损失越小。d越小,两个框的中心越接近,损失越小。
CompleteIOU损失
作者说,一个好的定位误差量化指标应该兼顾:重叠面积、中心点距离、长宽比。前两者在DIOU中都已经考虑了,因此CIOU又增加了长宽比这一因素,也就是说两个框的形状哪怕不全等,也应该尽可能相似。
$\alpha v$暂时看不懂是啥原理,暂时知道是在比较两个框的相似程度就够了,而且越小越相似。
本博客所有文章除特别声明外,均采用 CC BY-SA 4.0 协议 ,转载请注明出处!