MobileNet-SSD安装及配置

1 环境说明

  • OS: Ubuntu 16.04 LTS

  • Python 2.7.12(系统预装)

2 安装caffe-ssd

本部分参考caffe-ssd仓库README文档,安装caffe-ssd与安装caffe类似,可以参考caffe(CPU-only)安装及配置。

进入准备安装caffe-ssd的目录,执行以下命令获取其源码:

git clone https://github.com/weiliu89/caffe.git
cd caffe
git checkout ssd

进入caffe-ssd根目录,运行以下命令对源码进行编译:

# 根据自己的情况对Makefile.config进行修改。
cp Makefile.config.example Makefile.config
make -j8
# Make sure to include $CAFFE_ROOT/python to your PYTHONPATH.
make py
make test -j8
# (Optional)
make runtest -j8

确保CAFFE_ROOT/python包含在PYTHONPATH中,这里CAFFE_ROOT是指caffe-ssd的根目录:

export PYTHONPATH=/CAFFE_ROOT/python:$PYTHONPATH

4 下载MobileNet-SSD并运行demo

运行以下命令获取MobileNet-SSD的源码:

cd $CAFFE_ROOT/examples
git clone https://github.com/chuanqi305/MobileNet-SSD.git

下载预测网络模型MobileNetSSD_deploy.caffemodel放到MobileNet-SSD文件夹下,执行以下命令运行demo:

python ./demo.py

就可以看到预测结果了。

你可能感兴趣的:(caffe)