OneNet: Towards End-to-End One-Stage Object Detection

这篇paper的目的是实现不需要NMS的目标检测,这篇paper的主要做法在于更新 ground truth label assignment的方法。

目前主流算法的assignement方式用下图表示:

image

如果用一个损失函数来描述各自的Matching标准,则RetinaNet以及FCOS都是基于位置loss的,, 要么是中心点(FCOS),要么是IoU(RetinaNet),或者是两者的结合。

本文提出将分类loss也加到 matching cost上

且最终正样本只会分配给损失最小的那个样本 (minimum cost assignment)

image

结果常规,但是默认是Single head的

image

得到的网络速度相当快。