1、下载:git clone --recursive https://github.com/rbgirshick/py-faster-rcnn.git
2、配置Python的几个安装包
$ pip install cython
$ pip install easydict
$ sudo apt-get install python-opencv
3、Cython模块编译
在faster-rcnn的lib路径下
$ make
4、caffe和pycaffe的编译
复制$FRCN_ROOT/caffe-faste-rcnn的Makefile.config.example重命名为Makefile.config。
设置:把对应的注释符号记得删掉(#)
USE_CUDNN=1让CUDA支持DNN
WITH_PYTHON_LAYER=1让fasterRCNN支持Python接口。
# 现在改成(64位系统):
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/x86_64-linux-gnu /usr/lib/x86_64-linux-gnu/hdf5/serial
# 或者,你是32位系统:
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include /usr/include/hdf5/serial
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib /usr/lib/i386-linux-gnu /usr/lib/i386-linux-gnu/hdf5/serial/serial目的是为了找到hdf5.h
执行sed脚本:
find . -type f -exec sed -i -e 's^"hdf5.h"^"hdf5/serial/hdf5.h"^g' -e 's^"hdf5_hl.h"^"hdf5/serial/hdf5_hl.h"^g' '{}' \;
5、在$FRCN_ROOT/caffe-faste-rcnn下进行编译
在计算机搜索libhdf5_serial.so.10.1.0文件所在位置。然后在终端中输入:
sudo ln libhdf5_serial.so.10.1.0 libhdf5.so
sudo ln libhdf5_serial_hl.so.10.0.2 libhdf5_hl.so
sudo ldconfig (使得链接生效,make clean清除编译结果,查看opencv版本)
编译caffe相关文件 :~/caffe/src/caffe/proto$ protoc --cpp_out=/home/sjtu/lqq_learn_cnn/test/py-faster-rcnn/caffe-fast-rcnn/include/caffe/ caffe.proto
讲本机的caffe对应的cudnn文件覆盖过去:include里面的layers和util,以及src里面的layers和util里面对应的所有与cudnn相关的文件
make all -j4
6、在$FRCN_ROOT/caffe-faste-rcnn下继续编译
找到numpy文件的位置:
import numpy as np
np.get_include()
修改makefile文件:
PYTHON_INCLUDE:下面一句内容修改一下。
make pycaffe
7、下载demo模型数据()
下载模型
bash ./data/scripts/fetch_faster_rcnn_models.sh(失败了)
bash ./data/scripts/fetch_imagenet_models.sh(失败了)
解压模型:
tar xvf faster_rcnn_models.tgz
tar xvf imagenet_models.tgz
下载数据:
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtest_06-Nov-2007.tar
wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCdevkit_08-Jun-2007.tar
解压文件:
tar xvf VOCtrainval_06-Nov-2007.tar
tar xvf VOCtest_06-Nov-2007.tar
tar xvf VOCdevkit_08-Jun-2007.tar
对pascal voc2007 数据集做一个软链接:
ln -s VOCdevkit VOCdevkit2007
8、在$FRCN_ROOT/tool/下运行demo:
python demo.py
py_faster_rcnn配置
猜你喜欢
转载自blog.csdn.net/fanre/article/details/82908490
今日推荐
周排行