深度学习之目标检测(四)—— SSD算法
SSD算法
SSD, Single Shot MultiBox Detector(one-stage),Wei Liu在ECCV 2016上提出的一种目标检测算法
算法特点
- 直接回归目标类别和位置
- 不同尺度的特征图上进行预测
- 端到端的训练
- 图像分别率比较低,也能保证检测的精度
SSD网络结构
主干网络:VGGNet
最后通过NMS算法对检测结果进行合并和筛选
主干网络介绍
原作者采用的是vgg,但我们也可以采用resnet、mobilenet等网络结构
FC层即全连接层
多尺度FeatureMap预测
Prior Box Layer