Advertisement

Faster RCNN(caffe版本、Ubuntu系统)配置安装

阅读量:

网址:https://github.com/rbgirshick/py-faster-rcnn (rbg大神的caffe版本)

1. 下载代码

2. 新建文件夹FRCN_ROOT,将文件解压到FRCN_ROOT中

3. Build the Cython modules:

复制代码
    cd $FRCN_ROOT/lib
    make

4. Build Caffe and pycaffe:

删除FRCN_ROOT/caffe-faster-rcnn目录中所有文件(其实原也是空的);
https://github.com/rbgirshick/caffe-fast-rcnn/tree/0dcd397b29507b8314e252e850518c5695efbb83 上下载rbg大神的caffe版本,放入caffe-faster-rcnn目录下,原来下载的py-faster-rcnn压缩包中,没有将caffe-faster-rcnn目录中的一起下载下来;

复制代码
     拷贝未经编译过的最新版caffe(github上下载 https://github.com/BVLC/caffe )中所有文件到caffe-faster-rcnn中;
复制代码
    `cd ../FRCN_ROOT/caffe-fast-rcnn/`;
    `cd python/`;
    `for req in $(cat requirements.txt); do sudo pip install $req; done`(这句命令是为了安装一些必要的包等,需要联网);
    `cd ..` (返回到`FRCN_ROOT/caffe-fast-rcnn/` 目录下);
    `cp Makefile.config.example Makefile.config`(如果已经有配好caffe,可以把配好caffe中的Makefile.config文件直接复制过来,因为Makefile.config里有些参数需要注释修改的);

FRCN_ROOT/caffe-fast-rcnn/目录下所有与cudnn相关的文件都换成,能够编译成功的caffe版本中的cudnn相关文件;(因为rbg大神提供的caffe版本中的cudnn版本太低,和现在电脑上配的cudnn版本不相符):
包括目录:

复制代码
     /FRCN_ROOT/caffe-fast-rcnn/src/caffe/layers下的(16个文件)
    /FRCN_ROOT/caffe-fast-rcnn/include/caffe/util下的(一个:cudnn.hpp)
    /FRCN_ROOT/caffe-fast-rcnn/include/caffe/layers下的(8个)

依次跑通:

复制代码
    make all -j4  (如果这一步有报错,修改后,建议用 make clean ,然后重新来编译,不然以前出错后的编译文件在修改错误后可能继续造成其它错误,下面几步再出错就(应该)不用了)
    make test -j4
    make runtest -j4
    make pycaffe

5. 把下好的faster_rcnn_models放到py-faster-rcnn/data下

复制代码
    cd $FRCN_ROOT
    ./data/scripts/fetch_faster_rcnn_models.sh

6. 然后运行Demo:

复制代码
    cd $FRCN_ROOT
    ./tools/demo.py

全部评论 (0)

还没有任何评论哟~