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.  如图：

这个时候等结果吧！

结果：