windows py-faster-rcnn配置

  原创帖!转载请注明

    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

 

posted @ 2017-03-01 15:26  圆滚滚的小峰峰  阅读(10070)  评论(47编辑  收藏  举报