我的第一个tf-faster-rcnn训练程序(demo+voc2007数据训练)

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&

  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值