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. 如图:
这个时候等结果吧!
结果: