运行Faster-Rcnn代码
与Faster-Rcnn一样Faster-Rcnn官方也是采用caffe作为框架.
首先将项目克隆到本地(需要挂代理)
Make sure to clone with –recursive
git clone --recursive https://github.com/rbgirshick/py-faster-rcnn.git
然后进入lib目录执行make(如果是python3环境下 需要切换到Python2 修改make配置文件)
cd $FRCN_ROOT/lib
make
注意 faster-rcnn项目不能使用中文路径 ,把它放在home目录才能正常make
修改setup.py配置文件 仅CPU版本
由于我使用的是cpu版本,需要对makefile.config进行配置,然后 执行命令 make -j16
如果报错需要依赖,就需要安装和相应的依赖.
然后执行命令 make pycaffe
之后执行/data/scripts/fetch_faster_rcnn_models.sh 命令 下载模型文件,也可以用浏览器下载之后解压放在data文件夹里
最后执行命令 python2 ./demp.py –cpu
出错 就需要修改
../lib/fast_rcnn/nms_wrapper.py:9:#from nms.gpu_nms import gpu_nms
将gpu注释 仅使用cpu版本
在这里参考了 别人的博客 https://www.cnblogs.com/justinzhang/p/5386837.html
检测结果如下: