SSD: Single Shot MultiBox Detect

2020-10-13 本文已影响0人 ganyd

原作者：Wei Liu1, Dragomir Anguelov2等笔记整理：ganyongdong 2020.10.6
如有错误或者不清楚之处，欢迎指正交流，如需转载请注明出处

论文：https://arxiv.org/abs/1512.02325
代码：https://github.com/weiliu89/caffe/tree/ssd 推荐->https://github.com/lufficc/SSD.git
参考：https://zhuanlan.zhihu.com/p/79854543 https://zhuanlan.zhihu.com/p/65484308

摘要

设置默认框(default box)，类似于anchor box，具有不同的尺寸和宽高比。
联合多种分辨率下的特征图来检测不同尺寸的object，而不是直接对图像resize。
一阶段目标检测方法，实现了简单的端到端训练。
实验结果显示：在VOC2007，300x300输入，TitanX情况下，mAP=74.3%，FPS=59

网络结构

整个网络是由三大部分组成：

VGG Backbone
Extra Layers
Multi-box Layers

把vgg16后面3个全连接层替换成卷积层，用3x3卷积提取不同尺寸的特征图，在不同尺寸的特征图进行预测，之后用非极大值抑制得到结果。

默认框(default box)

在每个特征图的单元格的中心设置一系列尺度和大小不同的初始框，这些初始框都会反向映射到原图的某一个位置，如果某个初始框的位置正好和真实目标框的位置重叠度很高，那么就通过损失函数预测这个初始框的类别，同时对这些初始框的形状进行微调，以使其符合我们标记的真实目标框。
和yolo的anchor不同的是，ssd将其应用于几个不同分辨率的特征图，在多个特征图上使用不同形状的默认框。

ssd先验框.png

多尺度特征图

在backbone网络末端，替换成了卷积层，这些卷积层尺寸逐渐缩小，并允许网络在多个尺度上预测。

训练过程的匹配策略(默认框和真实标签框关联匹配)

将默认框与所有的真实标签框配对，只要两者之间的雅可比重叠值（jaccard overlap）大于某个阈值（比如 0.5）

损失函数

损失函数 $L$ 由位置损失 $L_{loc}$ 和分类置信度损失 $L_{conf}$ 的加权和成
$x_{ij}^p=\{1,0\}$ ：第i个默认框和第p类的第j个真实标签框相匹配的指示器，大于重叠阈值为1，小于重叠阈值为0

$L(x,c,l,g)=\frac{1}{N}(L_{conf}(x,c)+\alpha{L_{loc}(x,l,g)})$

其中：
$N$ ：已经匹配好的默认框的个数（如果N=0，就设置当前损失为0）
$l$ ：预测框和默认框之间的偏移
$g$ ：真实标签框和默认框之间的偏移

$L_{loc}(x,l,g)=\sum_{i\in{Pos}}^{N}\sum_{m\in\{cx,cy,w,h\}}x_{i,j}^ksmooth_{L1}(l_i^m-\hat{g}_j^m)$
$\hat{g}_j^{cx}=(g_j^{cx}-d_i^{cx})/d_i^w$
$\hat{g}_j^{cy}=(g_j^{cy}-d_i^{cy})/d_i^h$
$\hat{g}_j^w=log(\frac{g_j^w}{d_i^w})$
$\hat{g}_j^h=log(\frac{g_j^h}{d_i^h})$

其中：
$d$ ：默认框

$L_{conf}(x,c)=-\sum_{i\in{Pos}}^Nx_{ij}^plog(\hat{c}_i^p)-\sum _{i\in{Neg}}log(\hat{c}_i^0)$

$\hat{c}_i^p=\frac{exp(c_i^p)}{\sum_pexp(c_i^p)}$

选择默认框的尺寸和宽高比

对选择默认框有作者给出了一套推荐公式，但是针对不同数据集还是不同，如何最优是一个开放性的问题。

$S_k=S_{min}+\frac{S_{max}-S_{min}}{m-1}(k-1), k\in[1,m]$

其中， $S_{min}$ 是0.2， $S_{max}$ 是0.9。意味着最底层是0.2，最高层是0.9，期间所有大小按规律递增。

设定5种宽高比 $a_r\in\{1,\,2,\,3,\,0.5,\,0.3333\}$ ，则计算每个默认框的长度与宽度：
$w_k^a=s_k\sqrt{a_r}$
$h_k^a=s_k/\sqrt{a_r}$
对于长宽比为1的，另外增加一个默认框， $s_k^,=\sqrt{s_ks_{k+1}}$ ，所以特征图每个位置有6个默认框。