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

SSD算法

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

算法特点

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

 

深度学习之目标检测(四)—— SSD算法_第1张图片

 

SSD系列应用场景

深度学习之目标检测(四)—— SSD算法_第2张图片

 

SSD网络结构

深度学习之目标检测(四)—— SSD算法_第3张图片

主干网络:VGGNet

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

 

主干网络介绍

深度学习之目标检测(四)—— SSD算法_第4张图片

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

FC层即全连接层

 

多尺度FeatureMap预测

深度学习之目标检测(四)—— SSD算法_第5张图片

 

Prior Box Layer

深度学习之目标检测(四)—— SSD算法_第6张图片

深度学习之目标检测(四)—— SSD算法_第7张图片

得到6个不同的长宽比

深度学习之目标检测(四)—— SSD算法_第8张图片

深度学习之目标检测(四)—— SSD算法_第9张图片

 深度学习之目标检测(四)—— SSD算法_第10张图片

 

样本构造

深度学习之目标检测(四)—— SSD算法_第11张图片

 

数据增强

深度学习之目标检测(四)—— SSD算法_第12张图片

 

损失函数

深度学习之目标检测(四)—— SSD算法_第13张图片

深度学习之目标检测(四)—— SSD算法_第14张图片

 

SSD算法性能

深度学习之目标检测(四)—— SSD算法_第15张图片

深度学习之目标检测(四)—— SSD算法_第16张图片

深度学习之目标检测(四)—— SSD算法_第17张图片

 

SSD算法性能增强的原因

深度学习之目标检测(四)—— SSD算法_第18张图片

 

 

你可能感兴趣的:(深度学习,目标检测,SSD)