配置运行步骤
一、Git fast-rcnn源码
git clone –recursive https://github.com/rbgirshick/fast-rcnn.git
二、进fast-rcnn/lib目录进行编译
cd /home/服务器的名字/fast-rcnn/lib
make
目录如下:
三、编译
make -j16 && make pycaffe
可能出现的错误
1、cudnn.hpp
./include/caffe/util/cudnn.hpp:127:41: error: too few arguments to function ‘cudnnStatus_t cudnnSetPooling2dDescriptor(cudnnPoolingDescriptor_t, cudnnPoolingMode_t, cudnnNanPropagation_t, int, int, int, int, int, int)’
pad_h, pad_w, stride_h, stride_w));
^
解决办法:
1.cp caffe里的cudnn.hpp ./include/caffe/util/cudnn.hpp
2.将./src/caffe/layer里所有以cudnn开头的文件,如cudnn_xxx_layer.cu,cudnn_xxx_layer.cpp
都替换成最新版的caffe里的相应的同名文件。
2、cudnn_xxx_layer.hpp缺失
解决方法:将caffe/include中的layer文件夹复制到caffe-fast-rcnn中
四、运行
- 下载模型
./data/scripts/fetch_fast_rcnn_models.sh
2. 安装easydict
sudo apt-get install python-pip
sudo pip install easydict
3. 运行demo
./tools/demo.py