windows10下编译 fast rcnn

Windows下训练fast rcnn

一.准备

1.     caffe-master:windows的caffe安装包。https://github.com/Microsoft/caffe

这个模型对caffe有要求,需要roi_pooling这个层,所以你的去caffe/src/caffe/layers看下是否有着两个文件。如果没有这两个文件,后面会遇到错误,如图:


2.     我的caffe是配置的无GPU的,caffe配置可以看下我的博客。

http://blog.csdn.net/qq_34220460/article/details/59112649

3.     fast r-cnn:https://github.com/rbgirshick/fast-rcnn

4.     下载模型文件,也就是fast rcnn训练完后的模型文件。其实在fast rcnn文件中有脚本下载,但那是linux下的脚本,我们是在windows下。我给个别人的链接:http://pan.baidu.com/s/1kTzQYgR。下载完成后,请将3个文件夹放置在fast_rcnn/data中。

5.     然后是配置python接口,这个自己去配置下吧!

6.     打开cmd,输入sudo pip install cython和 sudo pipinstall easydict来安装cython和easydict,如果你使用的是Anaconda,cython默认是安装上了的。 
如果提示安装失败,可能是由于你的pip的版本过低,请先使用 
pip install pip --upgrade 
升级你的pip。

7.     还得再去安装opencv,去opencv官网下载并安装windows下的版本,链接:https://sourceforge.net/projects/opencvlibrary/files/opencv-win/3.2.0/opencv-3.2.0-vc14.exe/download

还要配置opencv的环境,将opencv的bin添加到系统环境变量中,最好看下链接,因为还有其他的要配置:http://jingyan.baidu.com/article/c45ad29c3e0e2d051753e204.html


不知这个,我们还要将opencv/build/python/2.7/x64 下的从cv2.pyd复制到python/libs/site-packages下,因为我们后面需要cv2这个模块,否则会报错,说no modules named cv2.

 

二.编译

  1.将caffe_master/Build/x64/Relase/pycaffe目录下的caffe ,复制到fast_rcnn/caffe-fast-rcnn中。

 2.用文本编辑器打开fast_rcnn/lib/utils/nms.pyx,将第25行的np.int_t修改为np.intp_t。

 3. 用文本编辑器打开fast_rcnn/lib/setup.py,将第18行和23行的 
"-Wno-cpp", "-Wno-unused-function"指令删除,只留下空的中括号[]即可。

4.  打开cmd,定位至fast_rcnn/lib目录中,执行pythonsetup.py install。 
如果提示Unable to find vcvarsall.bat的话,请输入以下指令: 
VS2012:SET VS90COMNTOOLS=%VS110COMNTOOLS% 
VS2013:SET VS90COMNTOOLS=%VS120COMNTOOLS% 
setup.py安装完成后,到python_root/Lib/site-packages/utils中可以找到两个文件cython_bbox.pyd和cython_nms.pyd,把这两个文件复制到fast_rcnn_root/lib/utils中。

5. 定位至fast_rcnn,执行python tools/demo.py,等待demo的结果吧!

 

三.问题解决

看似就上面这一点步骤,其实我们也是经过许多错误之后才总结出来的,我们也是有努力的。

1.     编译过程中,曾经遇到 no module named cython esaydict and cv2

这个就是上面你没有安装这些依赖包,在opecv的环境变量配置好了,可以到opencv\build\python\2.7\x64目录下看到cv2,如图:


2.     在此之后,就是上面所讲的错误。

如果提示Unable to find vcvarsall.bat的话,请输入以下指令: 
VS2012:SET VS90COMNTOOLS=%VS110COMNTOOLS% 
VS2013:SET VS90COMNTOOLS=%VS120COMNTOOLS% 

3.     就是遇到Check failed: registry.count(type) == 1 (0 vs. 1) Unknownlayer type: ROIPooling,如图:


,这个就有了前面为啥必须要有那两个文件,因为在windows下那两个文件没有被编译,所以我们需要把它加入到libcaffe下,cpp文件加入到src,。cu 文件加入到cu下去,还有就是加入头文件,把caffe目录下include/caffe/layers/roi_pooling_layer.hpp加入到include下,如图:


4.如果是no module named cv2, 请按照我们将的配置opencv那样配置,ok!!!

最后再把所有项目进行重新编译。

或许有的人做到了这一步,还是出错了,为什么呢?

或许你跟我一样,被当初的python配置坑了,当时测试python是否成功,不是有import caffe吗?import caffe成功就说明python接口配置成功,我当时是按照网上的一个办法,把这个 caffe master/build/x64/release/pycaffe下的caffe文件复制到了Python安装路径的Libs/site-packages下了,这样importcaffe成功了,但是没有想到的是后面加入了两个文件了,导致编译后的caffe 比之前的caffe多了roi_pooling layer,所以我们要重新把caffe复制到python安装路径的Libs/site-packages下,更新原来的caffe.  如图:


这个时候等结果吧!

结果: