个人微信公众号:AI研习图书馆,欢迎关注~
深度学习知识及资源分享,学习交流,共同进步~
1. 引言
SSD是一个端到端的网络,既借鉴了Faster RCNN中anchor boxes的概念、又借鉴了YOLO将检测转化为回归的思想,在它们的基础上,提出“从多层layer的feature maps上分别得到bounding box以实现多尺度检测”的思想,在速度和精度上都取得了卓越的表现。
现在,我们就在caffe框架下,来简单复现一下SSD网络。
2. 实现
2.1 caffe-ssd编译
首先,下载caffe-ssd,安装并编译。
https://github.com/weiliu89/caffe.git
git checkout ssd
跟编译caffe-master 是一样的,进入到caffe的根目录里:
cp Makefile.config.example Makefile.config
进入Makefile.config,设置如下,我们需要使用USE_CUDNN(前提你安装了CUDNN)
配置文件中其他地方的修改和caffe-master相同,相同机器可以直接拷贝使用,这里提供修改文件的教程,我的另一篇博客,和caffe一样修