CornerNet:一种anchor-free的目标检测算法.

模型介绍

作者提出了一个anchor-free的目标检测算法:CornerNet

CornerNetbackbone networkprediction module组成。

corner pooling

预测边界框的左上角坐标、右下角坐标,在预测点附近并没有出现物体,因此不能仅依赖预测点附近的感受野。作者提出了corner pooling,可以获取目标的边界信息。

top-left corner pooling为例,该点的结果由该点右边和下边所有特征点数值最大者决定:

corner pooling在网络中应用如下,网络还使用了残差连接:

损失函数

数据集在标注“Ground truth时并不是仅把目标边界框的左上角和右下角像素位置标注为1,而是使用一个二维高斯分布作为软标注:

网络的损失函数为:

\[L = L_{det} + αL_{pull} + βL_{push} + γL_{off}\]

\[L_{pull} = \frac{1}{N} \sum_{k=1}^{N} {[(e_{t_k}-e_k)^2+(e_{b_k}-e_k)^2]}\] \[L_{push} = \frac{1}{N(N-1)} \sum_{k=1}^{N} {\sum_{j=1,j≠k}^{N} {max(0,Δ-\mid e_k-e_j \mid)}}\] \[L_{off} = \frac{1}{N} \sum_{k=1}^{N} {\text{SmoothL1Loss}(o_k,\hat{o}_k)}\]