SSD(Single Shot MultiBox Detector)算是一个比较不错的目标检测算法,主攻方向是速度,当然精度也比Yolo提高了一些,最近在ubuntu16.04下实现了代码运行,此博文主要内容来自原作者的github,加上了一些个人理解,欢迎探讨。
PS:SSD代码和模型常常在更新,我给的链接可能不是最新版,如运行出错请参看官方github。
准备工作:linux+cuda+caffe是标配,我就不详述了,推荐博客:
ubuntu16.04+nvidia gt740m+cuda7.5+caffe安装、测试经历 http://blog.youkuaiyun.com/WoPawn/article/details/52313914
下面正式开始:
1.获取源码
git clone https://github.com/weiliu89/caffe.git
cd caffe
git checkout ssd
说明:SSD采用的是在caffe文件夹中内嵌例程的方式,作者改动了原版caffe,所以你需要把原来的caffe文件夹移除,git命令会新建一个带有SSD程序的caffe文件夹,当然,这个新的caffe要重新编译一次。
2.编译caffe
cd /home/mx/caffe
cp Makefile.config.example Makefile.config
运行时报错不断,事后总结,需要修改配置文件,用gedit或者vim打开配置文件进行修改:
1)Mak