下面主要针对 https://github.com/endernewton/tf-faster-rcnn工程,作者写的很好,但是对于我这种常识比较少的小白还是会遇到很多问题,所有都是从网上一点点找到的,整理在下面,希望可以减少大家的配置时间。
因为我是在windows10 subsystem ubuntu下跑的,所以没能用上GPU。下面我先按照我的配置说一遍,后面遇到什么问题再补充。
一、tf-faster rcnn 配置
1、下载代码: git clone https://github.com/endernewton/tf-faster-rcnn.git
2、由于运行在cpu上,所以要进行一些配置
a. 修改lib/setup.py文件,用‘return None’代替 ‘raise EnvironmentError’;注释掉Extension(‘nms.gpu_nms’…)整段代码。
b.修改lib/model/nms_wrapper.py文件,注释掉‘import gpu_nms’并且设置 ‘force_cpu = True’
3、build Cython modules
进入lib文件夹,执行 makeclean make 命令
4、下载VOC数据 https://github.com/rbgirshick/py-faster-rcnn#beyond-the-demo-installation-for-training-and-testing-models
5、下载与训练模型./data/scripts/fetch_faster_rcnn_models.sh
不知道是不是我网络的问题,我下载不下来,所以我通过文档中的链接下载的