目标检测笔记-SSD

SSD-经典的one stage

1.SSD结构

​

​ 上图所示的为SSD网络的结构图，前面的部分为VGG-16网络的一部分(对比上方的网络图可知)，输出卷积层4_3作为第一级的38*38大小的特征层输出，继续进行卷积处理分别得到19*19、10*10、5*5、3*3、1*1大小的特征层输出，共六个不同大小的输出，用来预测不同大小的物体，因为大的特征图细节更加丰富故，大的物体使用小的特征图进行预测，小的物体使用大的特征图预测，下面是对不同大小的特征图预测框的生成尺寸的选择。

​ 可以看到最终得到的预测框的数目有8732个，每个通过一个3*3*(Classes+4)大小的卷积核得到预测的种类的得分，以及预测框位置的回归参数。

2.正负样本的选取

正样本：1.用真实的目标框去匹配IOU最大的预测框；2.预测框与真实框的IOU值大于0.5。

负样本：对于剩下的负样本计算最大置信损失(highest confidence loss)，损失值越大证明越接近正样本，选取其中损失值最大的一些，使负样本与正样本的比例为3:1。

3.损失函数

​ 损失分为类别损失和定位损失两个部分，下面是SSD的类别损失，定位损失与Faster R-CNN的定位损失计算方法相同。