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