基于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