快速配置caffe-ssd及如何配置环境变量路径

caffe-ssd是目标检测的一个经典模型,属于单阶段检测模型,速度较快,本人实现过好几次,也实现过ssd的改进版本fssd和DSSD,效果能提升几个百分点。本篇博客主要介绍caffe-ssd的配置流程以及如何设置环境变量路径。

第一步:配置好caffe,包括NVIDIA-DIRVER,CUDA,CUDNN,OPENCV.可参考以下教程:

https://www.cnblogs.com/go-better/p/7161006.html

https://blog.csdn.net/yhaolpz/article/details/71375762

OK,以上工作做好下面开始配置caffe-ssd

第二步:

https://github.com/weiliu89/caffe/tree/ssd  

以上网址是ssd官网,可以直接下载也可以使用命令行:

git clone https://github.com/weiliu89/caffe/tree/ssd

接下来配置文件,可参考以下网址改:

https://blog.csdn.net/linhai1028/article/details/79452900

第三步:

make all -j8
make test
make runtest

涉及到权限问题可在每行命令行前加 sudo

编译成功再

sudo make pycaffe

成功就可以用了。

第四步:检查caffe是否配置成功

python
import caffe
exit()

这样表明caffe安装成功了。

第五步,caffe-ssd数据准备,可参照以下网址:

https://www.cnblogs.com/EstherLjy/p/6863890.html

第六步:设置环境变量路径:

sudo gedit ~/.bashrc
export PYTHONPATH=/caffe路径/python:$PYTHONPATH
source ~/.bashrc

在运行ssd_pascal.py时,出现如下错误:

网上查阅几乎所有的人都会说我这是环境变量没有设置好,可是我第六步明明做好了,百思不得其解,最后我打开ssd_pascal.py加入了两句话就ok了:

import sys
sys.path.append("caffe路径/python)

参考以下链接:

https://blog.csdn.net/silverdemon/article/details/77752873

至此,我的caffe-ssd就可以跑起来了!!

caffe-ssd预处理模型:

链接: https://pan.baidu.com/s/1frU5MPcNeg0GltgCrgTY9A 提取码: ifwh

猜你喜欢

转载自blog.csdn.net/qq_38522972/article/details/82888257