论文地址:Generalized Intersection over Union 算法
度量检测优劣基本基于 IOU,mAP 是典型的基于 IOU 的标准,可是 mAP 仅有一个 threshold,对于过了线的预测框一视同仁,不能进一步衡量其优劣,因此 MS COCO 挑战赛提出了多 IOU 阈值的综合 mAP 评价标准(就是同时采用几个阈值,计算出多个 mAP 综合打分)。ide
YOLO v1 直接回归 bbox 的位置参数(x,y,h,w),并在(h,w)采用预测平方根的方式回避尺度敏感函数
RCNN 预测的是相对预先检测出的候选区的位置偏移,采用 log 空间回避尺度敏感学习
此时的损失函数以 l2 为主优化
Faster RCNN 提出 l1 smooth 损失函数,使得学习鲁棒性更好spa
Faster RCNN 提出了 anchor boxes 取代了 RCNN 系列中的候选生成算法,可是过多的 anchor 候选框致使了正负样本不平衡的问题,其做者进一步提出了 focal loss 进行应对(存疑,通常来讲 one stage 法更易受到正负样本不平衡的困扰,而 two stage 法因为会在候选框阶段经过得分和nms筛选过滤掉大量的负样本,而后在分类回归阶段又固定正负样本比例,相较 one stage 法状况会好不少)。orm
对于同形状的 A 和 B,C 为相同形状的最小的能够包含 A、B 的图形(例如一般 A 和 B 是矩形框,此时 C 也取矩形框),下面公式给出 GIOU 的计算过程:blog
而后做者给出了 GIOU 的5个特色:继承
基于 min、max 和 分段函数的BP算法是可行的,因此 IOU 与 GIOU 的应用就不是问题了。做者说,虽然 IOU(GIOU) 计算不能获得解析解,但在限定了形状为矩形时就大大简化了问题:对角点进行 min、max 操做便可:get
做者还画了图试图证实 GIOU 损失函数的梯度相对于 IOU 损失函数更大,不过我感受有点牵强(没看懂),因此不贴了。
做者将 GIOU 损失函数应用到已有 2D 物体检测算法中,结果相对从前在两个指标上:IOU、GIOU 均有提升,毕竟优化目标就是检测指标,比起使用中间量的 l2 损失效果好是情理之中的。
文章源码尚未被释放,不过原理并不复杂,只要修改已有算法的损失函数便可,可是最近比较忙,没有时间实践一下了。我的以为是篇很中肯的文章,期待其余人复现实验验证一下,若是效果好说不定就像 BN、Relu 同样成为相关领域的标准配置了。