一文教你跑通Faster-Rcnn

第一步因为跑这个代码必须使用GPU进行加速,准备用CPU来跑的话,可以放弃了,安装cuda和cudnn,再安装NVIDIA的显卡驱动(版本一定要对),这方面网上教程非常之多,也有不少坑,仔仔细细一步一步的来是不会有多大问题的,安装完一定要进行一个测试,是否安装成功。

测试显卡驱动:$:nvidia-smi,出现GPU的使用情况,则代表成功


测试cuda:$:nvcc -V,出现Cuda的版本信息,则代表成功,cuda的环境变量一定要注意进行配置,写入/etc/profile

export PATH=/usr/local/cuda-8.0/bin:$PATH
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}} 
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}} 


第二步下载安装anaconda,下载python2.7版本的,后续我会提供3.5版本的faster-rcnn,下载地址:https://www.anaconda.com/download/

第三步下载Tensorflow版本的Faster-Rcnn代码,下载地址:https://github.com/CharlesShang/TFFRCNN

这个代码也并不是下载下来就能跑成功,需要填一些坑,按照Readme里的教程一步步来,基本没问题,总结起来就如下几个方面

1、数据集对应的下载下来放到对应位置

2、在lib下进行编译,生成gpu-nms和cpu-nms动态库,注意你的gcc版本,对make.sh文件进行相应的注释和解注,

尤其是要把-D_GLIBCXX_USE_CXX11_ABI=0改成-D_GLIBCXX_USE_CXX11_ABI=1,以及对psroipooling进行编译时也要加入-D_GLIBCXX_USE_CXX11_ABI=0。如下所示:



然后把train_net文件里的参数进行一个设置,即可运行


也许在这个过程中因人而异会出现不同的问题,但不要放弃,都可以解决,这个教程是比较傻瓜式的,因为我非常理解一个新手对于这个东西一无所知的时候,任何一个细节可能都需要琢磨半天。我会持续更新,有什么问题,欢迎留言讨论




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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值