ubuntu 14.04下 caffe环境中 faster rcnn安装与运行
本文是在caffe安装完成后的基础上安装fast rcnn,可以参考Windows与linux下caffe配置(仅CPU)
1.下载源码
git clone --recursive https://github.com/rbgirshick/py-faster-rcnn.git
2.生成Cython模块
cd /home/username/py-faster-rcnn/lib
make
cp Makefile.config.example Makefile.config
注: username为安装ubuntu的用户名
打开Makefile.config修改如下:(与配置caffe一样)
3.编译pycaffe
cd /home/username/py-faster-rcnn/caffe-fast-rcnn
make && make pycaffe
4.下载fast_rcnn训练好的模型
cd /home/username/py-faster-rcnn/data/scripts
./ fetch_faster_rcnn_models.sh
使用.sh下载,一般较慢,可以在这http://pan.baidu.com/s/1jIcfmrO 密码:l8nb下载,解压到py-faster-rcnn/data文件下即可
5.运行deme.py
修改/py-faster-rcnn/lib/fast_rcnn/nms_wrapper.py文件部分内容
#from nms.gpu_nms import gpu_nms
def nms(dets, thresh, force_cpu=True):
cd py-faster-rcnn/tools
./demo.py --cpu或者./demo.py --cpu --net zf
6.运行效果