深度学习之目标检测(四)—— SSD算法

SSD算法

SSD, Single Shot MultiBox Detector(one-stage),Wei Liu在ECCV 2016上提出的一种目标检测算法

算法特点

  • 直接回归目标类别和位置
  • 不同尺度的特征图上进行预测
  • 端到端的训练
  • 图像分别率比较低,也能保证检测的精度

 

SSD网络结构

深度学习之目标检测(四)—— SSD算法

主干网络:VGGNet

最后通过NMS算法对检测结果进行合并和筛选

 

主干网络介绍

深度学习之目标检测(四)—— SSD算法

原作者采用的是vgg,但我们也可以采用resnet、mobilenet等网络结构

FC层即全连接层

 

多尺度FeatureMap预测

深度学习之目标检测(四)—— SSD算法

 

Prior Box Layer

深度学习之目标检测(四)—— SSD算法

深度学习之目标检测(四)—— SSD算法