一文教你跑通Faster-Rcnn

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

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

一文教你跑通Faster-Rcnn_第1张图片

测试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文件里的参数进行一个设置,即可运行

一文教你跑通Faster-Rcnn_第2张图片

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




你可能感兴趣的:(人工只能,机器学习,faster-rcnn,rcnn,目标检测代码)