学习资料

论文

网络学习教程

开源SSD代码

  • PyTorch(笔者自己运行过,跑起来还算顺利)
  • TensorFlow
  • Caffe

模型

image.png
image.png

多尺度特征图检测

比较大的特征图来用来检测相对较小的目标,而小的特征图负责检测大目标

目标检测之-SSD - 图3

采用卷积进行检测

设置先验框

模型架构

VGG Backbone

根据 SSD 的论文描述,作者采用了 vgg16 的部分网络作为基础网络,在 5 层网络后,丢弃全连接,改为两个卷积网络,分别为:1024x3x3、1024x1x1。

目标检测之-SSD - 图4

Extra Layers

目标检测之-SSD - 图5

https://github.com/weiliu89/caffe/tree/ssd