【原创帖!转载请注明】
windows faster-rcnn安装一直会出现各种错误,本人在实验室电脑成功安装后,在自己电脑上同样的安装步骤也成功。所以做个总结,希望对大家有帮助。
一:安装环境
1、windows10
2、visual studio 2013
3、cuda7.5 for windows10
4、cudnn5.1 for windows 10
5、py-faster-rcnn https://github.com/rbgirshick/py-faster-rcnn
6、caffe https://github.com/Microsoft/caffe
7、py-faster-rcnn windows lib:https://github.com/MrGF/py-faster-rcnn-windows
具体安装文件见本人最后的资源链接。
二:软件安装
(1)CUDA7.5
cuda7.5直接安装操作,选择了默认安装路径。安装完成后,系统自动配置两个环境变量。
CUDA_PATH:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v7.5
CUDA_PATH_V7_5:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v7.5
为使用方便,我们在添加如下变量:
CUDA_BIN_PATH:%CUDA_PATH%\bin
CUDA_LIB_PATH:%CUDA_PATH%\lib\Win32
CUDA_SDK_BIN:%CUDA_SDK_PATH%\bin\Win64
CUDA_SDK_LIB:%CUDA_SDK_PATH%\common\lib\x64
CUDA_SDK_PATH:C:\ProgramData\NVIDIA Corporation\CUDA Samples\v7.5
测试cuda安装是否工程:
1、cmd输入nvcc -V
2、进入C:\ProgramData\NVIDIA Corporation\CUDA Samples\v7.5 点击Samples_vs2013.sln编译运行,出现如下图表面安装成功:
(2)cudnn5.1
下载cudnn5.1 解压得到include ,bin,lib三个文件夹,将三个文件夹复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v7.5下对应的include lib和bin文件夹下。
(3)Caffe安装
参考本人上篇博客,由于要是用faster-rcnn,所以编译的时候需要roi_pooling_layer.hpp,cu,cpp等加入到libcaffe文件配置中。具体参考http://www.cnblogs.com/LaplaceAkuir/p/6445189.html。
(4)py-faster-rcnn配置
下载py-faster-rcnn,建议git clone,不然缺少分支。确认caffe编译后的Build/x64/Release/添加到系统环境目录。
caffe编译下Build/x64/Release/pycaffe/caffe 目录复制到py-faster-rcnn/caffe-faster-rcnn/python目录下,替换原有文件。
安装环境7的py-faster-rcnn下的lib复制到py-faster-rcnn下替换到原来的lib文件。
安装必要库:conda install numpy pyqt ,本人用的anaconda2,可以直接安装。此步骤一般会安装多个依赖库。
cmd切换到py-faster-rcnn/lib目录:执行python setup.py install 成功后,修改setup_cuda.py中第33行,CUDA的include路径为你自己的路径。
执行 python setet_cuda.py install 安装成功后。
(5)运行demo
下载模型文件
方案1、官方链接:http://www.cs.berkeley.edu/~rbg/faster-rcnn-data下载faster -rcnn model
方案2、百度网盘地址:http://pan.baidu.com/s/1c1JQdby
cmd切到py-faster-rcnn目录下,执行python ./tool/demo.py 可能出现错误和解决办法:
1、编译遇到 AttributeError: ‘ProposalLayer’ object has no attribute ‘param_str_’ 问题,找到对应文件,修改param_str_为param_str ,完成编译。
2、proposal_layer.py出现 keyerror:’1’错误,将第64行改cfg_key='TEST'#str(self.phase)
至此,编译完成,运行成功。
三:安装文件资源
cuda7.5.18 ,cudnn5.1: http://pan.baidu.com/s/1dE6qKFn
faster-rcnn model: http://pan.baidu.com/s/1c1JQdby
py-faster-rcnn for lib: https://github.com/MrGF/py-faster-rcnn-windows