在Windows环境下搭建py-faster-rcnn:
【0.参考资料】:
在搭建过程用遇到不少问题,感谢下面的参考资料给予的帮助。
- happynear的blog: http://blog.youkuaiyun.com/happynear/article/details/46822109#comments
- github: https://github.com/MrGF/py-faster-rcnn-windows
- 程序园:http://www.voidcn.com/blog/wl2002200/article/p-6209599.html
- windows-caffe:http://blog.youkuaiyun.com/sunmelon/article/details/54600178
按前辈的话说,faster还是matlab版本比较好配置,可我却偏偏钟爱python,即使再难也克服了。下面说一下过程:
【windows-caffe】:
我用的VS2013,caffe是MS维护的版本,有点略坑,至于怎么编译MS的caffe,我就不说了,网上有很多。
网址:https://github.com/Microsoft/caffe,用他的原因是因为他包含现在比较流行的如faster,ssd等自己写的网络层,而不用像Ubuntu下面那样编译caffe。
在编译windows-caffe的时候有个小问题,默认是没有把roi_pooling层放进去编译的,这样的后果就是之后在运行网络的时候会出现Check failed: registry.count(type) == 1 (0 vs. 1) Unknown layer ROIPooling的问题。
把roi_pooling_layer.hpp , roi_pooling_layer.cpp , roi_pooling_layer.cu 添加到libcaffe项目的layer层下,具体步骤参照参考资料4.在添加完以后重新编译,最后编译出来的pycaffe的接口是在Build/x64/Release/pycaffe/caffe目录下。
【Faster rcnn】
- 下载Faster rcnn的源码(地址为https://github.com/rbgirshick/py-faster-rcnn)。
- 下载github上的lib来替换原来Faster根目录下的lib文件夹: https://github.com/MrGF/py-faster-rcnn-windows
- 用命令行(CMD)到{Faster root} / lib ,然后执行
python setup.py (cpu) python setup_cuda.py (GPU)
- 至此,所需要的东西都编译好了,可以执行 python tools/demo.py了
【可能遇到的问题】
在运行的时候可能会出现proposal_layer.py出现 keyerror:’1’错误,我运行的时候是将文件中的第64行改成:cfg_key = ‘TEST’,然后就可以了