《CornerNet：DetectingObjectsasPairedKeypoints》论文笔记

这篇文章提出了CornerNet的目标检测模型&＃xff0c;该模型是一个直通检测检测模型&＃xff0c;与Faster RCNN那种两阶段的检测模型不一样&＃xff0c;更类似于SSD与YOLO。但是与SSD等模型不同的是其并不使用anchor boxes&＃xff0c;而是提出了一种叫做corner pooling的操作帮助网络更好地定位目标。在COCO数据集上的表现性能超过了所有现有的检测网路。
anchor boxes机制在现有两阶段以及直通检测网络模型中被广泛使用&＃xff0c;这些模型中也是通过anchor的检出来进行边界的回归。但是anchor也会带来其的缺点&＃xff0c;该文章中大体给出了两点&＃xff1a;
1&＃xff09;通常需要大量的anchor&＃xff0c;在DSSD中需要40k&＃xff0c;而在RetinaNet中则需要100k。这么大量的anchor就带来了正例anchor的数量与负例anchor的比例及其不成比例&＃xff0c;从而降低了网络的训练速度。
2&＃xff09;anchor boxes的引入带来了许多的超参数&＃xff0c;并且需要进行细致设计。包括anchor boxes的数量、尺寸、长宽比例。特别是在单一网络在多尺度进行预测的情况下会变得复杂&＃xff0c;每个尺度都需要独立设计。
下图是这篇文章中使用的网络结构可以描述为&＃xff1a;

在该方法中舍弃传统的anchor boxes思路&＃xff0c;提出CornerNet模型预测目标边界框的左上角和右下角一对顶点。使用单一卷积模型生成热点图和连接矢量&＃xff1a;所有目标的左上角和所有目标的右下角热点图&＃xff0c;每个顶点的连接矢量(embedding vector)。也就是该方法在卷积的输出结果上&＃xff0c;分别预测边界框的左上角与右下角&＃xff0c;之后寻找它们之间的对应关系。

主要思路其实来源于一篇多人姿态估计的论文。基于CNN的2D多人姿态估计方法&＃xff0c;通常有2个思路&＃xff08;Bottom-Up Approaches和Top-Down Approaches&＃xff09;&＃xff1a;
Top-Down framework&＃xff1a;就是先进行行人检测&＃xff0c;得到边界框&＃xff0c;然后在每一个边界框中检测人体关键点&＃xff0c;连接成每个人的姿态&＃xff0c;缺点是受人体检测框影响较大&＃xff0c;代表算法有RMPE&＃xff1b;
Bottom-Up framework&＃xff1a;就是先对整个图片进行每个人体关键点部件的检测&＃xff0c;再将检测到的人体部位拼接成每个人的姿态&＃xff0c;代表方法就是openpose。

在文章中给出了两个创新点&＃xff1a;
1&＃xff09;文章将目标检测上升到方法论&＃xff0c;基于多人姿态估计的Bottom-Up思想&＃xff0c;首先同时预测定位框的顶点对&＃xff08;左上角和右下角&＃xff09;热点图和embedding vector&＃xff0c;根据embedding vector对顶点进行分组&＃xff0c;论文中提到分组的依据是同目标的两个角点距离是很小的。
2&＃xff09;文章提出了corner pooling用于定位顶点。自然界的大部分目标是没有边界框也不会有矩形的顶点&＃xff0c;依top-left corner pooling 为例&＃xff0c;对每个channel&＃xff0c;分别提取特征图对应位置处的水平和垂直方向的最大值&＃xff0c;然后求和。

3&＃xff09;文中提出的CornerNet模型基于hourglass架构&＃xff0c;使用focal loss的变体训练神经网络。

为什么基于角点的检测要比基于anchor boxes的检测效果更好&＃xff0c;这里假设了两个原因&＃xff1a;
1&＃xff09;anchor boxes的中心是依赖于目标的四条边&＃xff0c;但是角点却只需要两个边&＃xff0c;而且还是使用了corner pooling&＃xff0c;因而表现比anchor好。
2&＃xff09;采用了更加高效的空间检测框机制&＃xff0c;这里只使用$O(w\ast h)$个角点就代表了O(W2∗h2)O(W^2*h^2)O(W2∗h2)的可能检测框。

CornerNet的整体结构&＃xff1a;

上面的图中给出的是CornerNet的网络结构图&＃xff0c;可以看出CornerNet模型架构包含三部分&＃xff0c;Hourglass Network&＃xff0c;Bottom-right corners&Top-left Corners Heatmaps和Prediction Module。
其中Hourglass Network是人体姿态估计的典型架构&＃xff0c;论文堆叠两个Hourglass Network生成Top-left和Bottom-right corners&＃xff0c;每一个corners都包括corners Pooling&＃xff0c;以及对应的Heatmaps, Embeddings vector和offsets&＃xff08;为了提升框的精度&＃xff09;。embedding vector使相同目标的两个顶点&＃xff08;左上角和右下角&＃xff09;距离最短&＃xff0c; offsets用于调整生成更加紧密的边界定位框。最后预测框也是在这三个生成的结果的基础上使用后处理算法进行实现的。

2.1 Detecting Corners

Paper模型生成的heatmaps包含C channels&＃xff08;C是目标的类别&＃xff0c;没有background channel&＃xff09;&＃xff0c;每个channel是二进制掩膜&＃xff0c;表示相应类别的顶点位置。

对于每个顶点&＃xff0c;只有一个ground-truth&＃xff0c;其他位置都是负样本。但是网络的输出并不会从开始就与上图标注的绿色虚线重合&＃xff0c;因而是按照了一定的策略来进行组合的。具体来讲是这样的&＃xff1a;在训练过程&＃xff0c;模型减少负样本&＃xff0c;在每个ground-truth顶点设定半径r区域内都是正样本&＃xff0c;这是因为落在半径r区域内的顶点依然可以生成有效的边界定位框&＃xff0c;论文中设置$IoU\&＃61;0.7$。那么对于与实际的定位有偏移的网络输出论文使用了惩罚&＃xff0c;是按照没有归一化的而为高斯函数e−x2&＃43;y22σ2e^{-\frac{x^2&＃43;y^2}{2\sigma^2}}e−2σ2x2&＃43;y2​设置的&＃xff0c;其中σ&＃61;13\sigma&＃61;\frac{1}{3}σ&＃61;31​&＃xff0c;中心坐标是标注的角点定位。
pcijp_{cij}pcij​表示类别为c&＃xff0c;坐标是$(i,j)$的预测热点图&＃xff0c;ycijy_{cij}ycij​表示相应位置的ground-truth&＃xff0c;其指融合了前面的提到的高斯半径。论文提出变体Focal loss表示检测目标的损失函数&＃xff1a;

其中参数$N$是图片中目标的总数&＃xff0c;参数$\alpha \&＃61;2$与$\beta \&＃61;4$适用于控制每个部分的贡献程度。
在卷积神经网络中存在着下采样层&＃xff0c;这样从原始的图像输入到最后的heatmap产生的这个过程会累计误差&＃xff0c;特别是在对一些小目标的物体进行检测的时候&＃xff0c;这样的误差就无法接受了&＃xff0c;因而文章引入了偏移修正来修正它&＃xff1a;

然后使用L1损失与原图标注的位置对其进行修正&＃xff1a;

最后的网络损失函数是上面的几个相加的形式&＃xff1a;

其中&＃xff0c;$\alpha \&＃61;\beta \&＃61;0.1,\gamma \&＃61;1$。

2.2 Grouping Corners

输入图像会有多个目标&＃xff0c;相应生成多个目标的左上角和右下角顶点。对顶点进行分组&＃xff0c;论文引入Associative Embedding的思想&＃xff0c;模型在训练阶段为每个corner预测相应的embedding vector&＃xff0c;通过embedding vector使同一目标的顶点对距离最短&＃xff0c;即是模型可以通过embedding vector为每个顶点分组。
模型训练Lpull损失函数L_{pull}损失函数Lpull​损失函数使同一目标的顶点进行分组&＃xff0c; Lpush损失函数L_{push}损失函数Lpush​损失函数用于分离不同目标的顶点。etke_{tk}etk​是左下角的顶点&＃xff0c;ebke_{bk}ebk​是右下角的顶点。

2.3 Corner Pooling

2.3.1 top-left corner

一般来说在heatmap上并不存在直观的选择依据&＃xff0c;因而文章中采用了corner pooling的方法。对于在左上角的角点&＃xff08;顶点&＃xff09;使用其在水平与垂直方向上分分别使用max pooling&＃xff0c;然后对其结果合并来实现角点的定位&＃xff0c;其运算流程可以通过下面这幅图来进行说明

对于要判定点$(x,y)$是不是左上角的角点&＃xff0c; 这里使用ftf_tft​与flf_lfl​去代表这个左上角点pooling layer的特征图&＃xff0c;使用fti,jf_{t_{i,j}}fti,j​​与fli,jf_{l_{i,j}}fli,j​​来代表在位置$(i,j)$处ftf_tft​与flf_lfl​的特征向量。对于输入大小为$W\ast H$大小的特征图&＃xff0c;计算位置$(i,j)$处到$(i,H)$在ftf_tft​的向量ti,jt_{i,j}ti,j​&＃xff0c;计算位置$(i,j)$处到$(j,W)$在flf_lfl​的向量li,jl_{i,j}li,j​&＃xff0c;最后再将两个加起来作为最后的输出。

2.3.2 bottom-right corner

bottom-right corner的计算与上面top-left的计算是类似的。它在$(0,j)$到$(i,j)$的特征图与$(i,0)$到$(i,j)$的特征图上进行最大池化操作之后再将两者得到的结果相加得到最后的结果。

2.3.3 预测模块

预测模块使用的结构使用ResNet与卷积组合的形式&＃xff0c;其结构&＃xff08;top-left corner部分&＃xff09;如下。最后预测生成Heatmap、Embeddings、Offsset。

2.4 Hourglass Network

Hourglass Network同时包含了bottom-up&＃xff08;from high resolutions to low resolutions)和top-down (from low resolutions to high resolutions)。而且&＃xff0c;整个网络有多个bottom-up和top-down过程。这样设计的目的是在各个尺度下抓取信息。针对目标检测任务&＃xff0c;论文调整了Hourglass一些策略。

**PS:**在这篇文章涉及到了SHM&＃xff08;Stack Hourglass Model&＃xff09;&＃xff0c;不了解的小伙伴可以参考这篇文章快速了解&＃xff1a;Stacked Hourglass Networks - 堆叠沙漏网络结构详解