安装caffe之前,你需要先做一些准备工作,包括:
- 安装caffe的依赖包
- 安装cuda
- 安装opencv
关于这些步骤,网上已经有很多写得非常详细的教程,在此就不多赘述了。读者可以参考一下这些博文《Ubuntu16.04 14.04安装配置Caffe(GPU版)——天梦冰蚕》、《Ubuntu16.04 Caffe 安装步骤记录(超详尽)——王英豪》
这里主要介绍一下如何配置caffe
下载caffe-ssd源码
由于MobileNetv2-SSDlite需要使用ssd版本的caffe,所以我们需要到github上下载对应的源码
git clone https://github.com/weiliu89/caffe.git
cd caffe
若你要使用tensorflow提供的pre-trained model那么你需要安装chuanqi305提供的caffe-ssd。这个版本在weiliu89版本的基础上添加了ReLU6 layer
本文以weiliu89版本的caffe为例子
git clone https://github.com/chuanqi305/ssd.git
mv ssd caffe
cd caffe
制作Makefile.config文件
首先,拷贝caffe提供的example文件
cp M