1. 本人系统配置:
LINUX+python3.6+tensorflow+cpu
(主要参考:https://blog.csdn.net/char_QwQ/article/details/80980505)
2. 克隆代码
git clone https://github.com/endernewton/tf-faster-rcnn.git
3. 根据显卡或cpu更改对应的架构
(参考:https://blog.csdn.net/sinat_33486980/article/details/81045315)
3.1修改lib/model/nms_wrapper.py:
按照图中箭头所指示的地方,将代码注释,作用就是为了直接禁用掉GPU模式
3.2lib/setup.py
3.3将USE_GPU_NMS 由原来的True改为False
4. 编译Cython
仍然在lib路径下,编译Cython模块(确保你已经安装了easydict&