第一部分 SSD安装
系统:ubuntu 14.04
语言:python
ssd项目主页:https://github.com/weiliu89/caffe/tree/ssd
首先,我们把项目代码clone下来, 然后编译:
git clone https://github.com/weiliu89/caffe.git
cd caffe
git checkout ssd
接下来,我们开始编译caffe,编译caffe非常容易,此篇讲的是caffe + cpu + ubuntu14.04的安装, GPU版安装需修改Makefile.config文件, 修改完成后:
make
make py
到这里我们就完成了SSD的安装,接下来我们讲一下如何训练自己的数据集。
第二部分 训练自己的数据集
首先我们不妨先跑一下项目的demo, 需要下载数据集,提前训练好的数据集等。
下载预训练的模型,链接:https://gist.github.com/weiliu89/2ed6e13bfd5b57cf81d6,下载完成后保存在:
caffe/models/VGGNet/
下载VOC2007和VOC2012数据集, 放在/data目录下:

最低0.47元/天 解锁文章
897





