SSD算法

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

算法特点

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

SSD网络结构

 

主干网络:VGGNet

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

主干网络介绍

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

FC层即全连接层

多尺度FeatureMap预测

Prior Box Layer

得到6个不同的长宽比

 

样本构造

数据增强

损失函数

SSD算法性能

SSD算法性能增强的原因

更多推荐