faster rcnn训练(一)跑通原生模型

基于ubuntu16.04  caffe  python

一、下载文件

在Ubuntu根目下运行命令行的下载:

  git clone --recursive https://github.com/rbgirshick/py-faster-rcnn.git

二、编译caffe

1.进入py-faster-rcnn/lib   make

2.进入py-faster-rcnn/caffe-fast-rcnn    复制 Makefile.config.example       Rename为       Makefile.config

修改参数

 
 
WITH_PYTHON_LAYER := 1
USE_CUDNN := 1

3.由于caffe版本不同的问题,将caffe下的文件拷贝或替换入caffe-faster-rcnn:

include/caffe/util/cudnn.hpp 

include/caffe/layers  所有以cudnn开头的文件

src/caffe/layer  所有以cudnn开头的文件

4.编译caffe

进入py-faster-rcnn/caffe-faster-rcnn

  make -j64


编译成功后

make pycaffe

三、下载训练好的模型

1.进入py-faster-rcnn  下载faster-rcnn models trained

./data/scripts/fetch_fater_rcnn_models.sh

2.下载caffe models(ZF、VGG16等)

./data/scripts/fetch_imagenet_modelsl.sh

四、跑通demo.py

进入py-faster-rcnn

./tools/demo.py  - - gpu 0





猜你喜欢

转载自blog.csdn.net/weixin_39732260/article/details/80929504
今日推荐