模型简介
SSD,全称Single Shot MultiBox Detector,single shot指的是SSD算法属于one-stage方法,MultiBox说明SSD是多框预测。SSD算法的性能与速度优于YOLO。
模型结构
SSD模型的输入图片大小是300x300,或512x512。
SSD的主干特征提取网络依据VGG-16,并选取了6个有效特征层用来产生先验框,SSD采用了不同尺度和长宽比的先验框,每个网格的先验框的数量有4和6,最终共产生8732个box,进行分类和回归。
anchor
default box的尺度计算方式如下:
Loss函数
Loss = softmax_loss + l1_smooth_loss