- 什么是MobileNet-SSD
SSD本身就是一个基于VGG16的网络,MobileNet-SSD 顾名思义就是这两个网络的组合,具体怎样组合详见网络结构图,下面详细介绍SSD的算法原理及SSD的结构。
- SSD算法的主要网络结构是VGG16,只是将最后两个全连层改成卷积层,并又增加了4个卷积层来构造网络结构。
- 对5种不同的卷积层的输出(feature map)分别用两个不同的3x3卷积核进行卷积,一个输出分类用的confidence,每个default box(实际上是prior box)生成的21个类别(在VOC数据集上的训练为例)的confidence,另一个输出回归用的location,每个box生成4个坐标值(x,y,w,h)。
Priorbox:目标的预选框,得到预选框,后续就可以通过softmax分类
- 这5种feature map还经过priorbox 层生成的 priorbox(即坐标)。每个feature map层中的default box的数量在caffe 中是给定的即8732个,计算如下:
38x38x4+19x19x6+10x10x6+5x5x6+3x3x4+1X1X4=8732。最后将前面三个计算结构合并,然后传给loss层。