Faster R-CNN 安装以及demo的测试

最新推荐文章于 2023-04-04 17:42:15 发布

Pengfei_Chen

最新推荐文章于 2023-04-04 17:42:15 发布

阅读量573

点赞数

分类专栏：深度学习文章标签：深度学习 Faster RCNN

本文链接：https://blog.csdn.net/Pengfei_Chen/article/details/79988991

版权

深度学习专栏收录该内容

2 篇文章 0 订阅

订阅专栏

今天开始写下我的第一篇CSDN博客，关于在ubantu上安装Faster R-CNN以及一些编译问题的解决方式，最后通过运行测试源码中的demo.py程序来验证安装成功。

在下载源码之前，先看一下自己电脑的硬件配置，我用的电脑配置如下，简单罗列一下CPU，内存以及GPU的信息。

model name      : Intel(R) Xeon(R) CPU E5-2630 0 @ 2.30GHz
model name      : Intel(R) Xeon(R) CPU E5-2630 0 @ 2.30GHz
model name      : Intel(R) Xeon(R) CPU E5-2630 0 @ 2.30GHz
model name      : Intel(R) Xeon(R) CPU E5-2630 0 @ 2.30GHz
model name      : Intel(R) Xeon(R) CPU E5-2630 0 @ 2.30GHz
model name      : Intel(R) Xeon(R) CPU E5-2630 0 @ 2.30GHz
model name      : Intel(R) Xeon(R) CPU E5-2630 0 @ 2.30GHz
model name      : Intel(R) Xeon(R) CPU E5-2630 0 @ 2.30GHz
model name      : Intel(R) Xeon(R) CPU E5-2630 0 @ 2.30GHz
model name      : Intel(R) Xeon(R) CPU E5-2630 0 @ 2.30GHz
model name      : Intel(R) Xeon(R) CPU E5-2630 0 @ 2.30GHz
model name      : Intel(R) Xeon(R) CPU E5-2630 0 @ 2.30GHz
MemTotal:       32893492 kB
05:00.0 VGA compatible controller: NVIDIA Corporation GF108GL [Quadro 600] (rev a1)
05:00.1 Audio device: NVIDIA Corporation GF108 High Definition Audio Controller (rev a1)
06:00.0 VGA compatible controller: NVIDIA Corporation Device 1b06 (rev a1)
06:00.1 Audio device: NVIDIA Corporation Device 10ef (rev a1)

好了，那么现在可以开始从GitHub上下载Faster R-CNN的源码了。在终端先输入:

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

注意一定要加上参数--recursive，这样会给后面带来很多方便，下载好之后我们需要先编译cython。

首先进入主目录下的lib文件夹，然后执行make操作：

cd py-faster-rcnn/lib
make

这个时候会出现出现报错：

ModuleNotFoundError: No module named 'Cython'

说明没有安装‘Cython’模块，我们可使用pip进行安装，同时我们还需要两个模块，一个是easydict，另一个是python-opencv，前两个模块我都是使用pip进行安装，最后一个模块我使用的是conda进行安装，当然python-opencv也可以使用源码进行安装。安装代码如下：

pip install cython
pip install easydict
install --channel https://conda.anaconda.org/menpo opencv3

make操作完成之后，我们要开始编译caffe和pycaffe，首先进入py-faster-rcnn/caffe-fast-rcnn目录，我们修改一下Makefile.config文件，首先执行以下操作：

cp Makefile.config.example Makefile.config 
vim Makefile.config

由于我使用的python环境是anaconda创建的python2.7环境，所以我自己的makefile.config文件如下所示（这部分仅供参考，不同的环境配置可以使makefile.config不同），但是一定要做的是将USE_CUDNN := 1和WITH_PYTHON_LAYER := 1前面的#去掉。

## Refer to http://caffe.berkeleyvision.org/installation.html

# Contributions simplifying and improving our build system are welcome!

# cuDNN acceleration switch (uncomment to build with cuDNN).
USE_CUDNN := 1

# CPU-only switch (uncomment to build without GPU support).
# CPU_ONLY := 1

# uncomment to disable IO dependencies and corresponding data layers
# USE_OPENCV := 0
# USE_LEVELDB := 0
# USE_LMDB := 0

# uncomment to allow MDB_NOLOCK when reading LMDB files (only if necessary)
#	You should not set this flag if you will be reading LMDBs with any
#	possibility of simultaneous read and write
# ALLOW_LMDB_NOLOCK := 1

# Uncomment if you're using OpenCV 3
# OPENCV_VERSION := 3

# To customize your choice of compiler, uncomment and set the following.
# N.B. the default for Linux is g++ and the default for OSX is clang++
# CUSTOM_CXX := g++

# CUDA directory contains bin/ and lib/ directories that we need.
CUDA_DIR := /usr/local/cuda
# On Ubuntu 14.04, if cuda tools are installed via
# "sudo apt-get install nvidia-cuda-toolkit" then use this instead:
# CUDA_DIR := /usr

# CUDA architecture setting: going with all of them.
# For CUDA < 6.0, comment the *_50 through *_61 lines for compatibility.
# For CUDA < 8.0, comment the *_60 and *_61 lines for compatibility.
# For CUDA >= 9.0, comment the *_20 and *_21 lines for compatibility.
CUDA_ARCH := -gencode arch=compute_20,code=sm_20 \
		-gencode arch=compute_20,code=sm_21 \
		-gencode arch=compute_30,code=sm_30 \
		-gencode arch=compute_35,code=sm_35 \
		-gencode arch=compute_50,code=sm_50 \
		-gencode arch=compute_52,code=sm_52 \
		-gencode arch=compute_60,code=sm_60 \
		-gencode arch=compute_61,code=sm_61 \
		-gencode arch=compute_61,code=compute_61

# BLAS choice:
# atlas for ATLAS (default)
# mkl for MKL
# open for OpenBlas
BLAS := atlas
# Custom (MKL/ATLAS/OpenBLAS) include and lib directories.
# Leave commented to accept the defaults for your choice of BLAS
# (which should work)!
# BLAS_INCLUDE := /path/to/your/blas
# BLAS_LIB := /path/to/your/blas

# Homebrew puts openblas in a directory that is not on the standard search path
# BLAS_INCLUDE := $(shell brew --prefix openblas)/include
# BLAS_LIB := $(shell brew --prefix openblas)/lib

# This is required only if you will compile the matlab interface.
# MATLAB directory should contain the mex binary in /bin.
# MATLAB_DIR := /usr/local
# MATLAB_DIR := /Applications/MATLAB_R2012b.app

# NOTE: this is required only if you will compile the python interface.
# We need to be able to find Python.h and numpy/arrayobject.h.
#PYTHON_INCLUDE := /usr/include/python2.7 \
		/usr/lib/python2.7/dist-packages/numpy/core/include
# Anaconda Python distribution is quite popular. Include path:
# Verify anaconda location, sometimes it's in root.
ANACONDA_HOME := /home/brick/anaconda3/envs/py27
PYTHON_INCLUDE := $(ANACONDA_HOME)/include \
		$(ANACONDA_HOME)/include/python2.7 \
		$(ANACONDA_HOME)/lib/python2.7/site-packages/numpy/core/include

# Uncomment to use Python 3 (default is Python 2)
# PYTHON_LIBRARIES := boost_python3 python3.5m
# PYTHON_INCLUDE := /usr/include/python3.5m \
#                 /usr/lib/python3.5/dist-packages/numpy/core/include

# We need to be able to find libpythonX.X.so or .dylib.
#PYTHON_LIB := /usr/lib
PYTHON_LIB := $(ANACONDA_HOME)/lib

# Homebrew installs numpy in a non standard path (keg only)
# PYTHON_INCLUDE += $(dir $(shell python -c 'import numpy.core; print(numpy.core.__file__)'))/include
# PYTHON_LIB += $(shell brew --prefix numpy)/lib

# Uncomment to support layers written in Python (will link against Python libs)
WITH_PYTHON_LAYER := 1

# Whatever else you find you need goes here.
INCLUDE_DIRS := $(PYTHON_INCLUDE) /usr/local/include
LIBRARY_DIRS := $(PYTHON_LIB) /usr/local/lib /usr/lib

# If Homebrew is installed at a non standard location (for example your home directory) and you use it for general dependencies
# INCLUDE_DIRS += $(shell brew --prefix)/include
# LIBRARY_DIRS += $(shell brew --prefix)/lib

# NCCL acceleration switch (uncomment to build with NCCL)
# https://github.com/NVIDIA/nccl (last tested version: v1.2.3-1+cuda8.0)
# USE_NCCL := 1

# Uncomment to use `pkg-config` to specify OpenCV library paths.
# (Usually not necessary -- OpenCV libraries are normally installed in one of the above $LIBRARY_DIRS.)
# USE_PKG_CONFIG := 1

# N.B. both build and distribute dirs are cleared on `make clean`
BUILD_DIR := build
DISTRIBUTE_DIR := distribute

# Uncomment for debugging. Does not work on OSX due to https://github.com/BVLC/caffe/issues/171
# DEBUG := 1

# The ID of the GPU that 'make runtest' will use to run unit tests.
TEST_GPUID := 0

# enable pretty build (comment to see full commands)
Q ?= @

makefile.config文件修改好之后，我们在终端输入

make -j8 && make pycaffe

下面来分享一下我在编译过程中我遇到的各种bug，希望能够给大家避避雷区。

1.和hdf5.h文件相关的问题

打开Makefile文件，找到以下行，将内容进行替换：

LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_hl hdf5
修改成：
LIBRARIES += glog gflags protobuf boost_system boost_filesystem m hdf5_serial_hl hdf5_serial

2.和imgcodecs.so有关的问题

打开Makefile问件，找到以下行，将内容进行替换：

LIBRARIES += opencv_core opencv_highgui opencv_imgproc
修改成：
LIBRARIES += opencv_core opencv_highgui opencv_imgproc opencv_imgcodecs

3.和cudnn与caffe版本不匹配相关问题，报错大概是too few arguments ....

这个是由于cudnn版本太高导致，网上有些方法说是将caffe-fast-rcnn/include里面的一些文件替换成/caffe-fast-rcnn/src中的一些文件，但是我试了之后发现并没有效果，通过继续查询资料我得到了一个较好的办法，在GitHub上继续下载一些文件进行更新。

1）

cd caffe-fast-rcnn
git remote add caffe https://github.com/BVLC/caffe.git
git fetch caffe
git merge -X theirs caffe/master

2）remove self_.attr("phase") = static_cast<int>(this->phase_); from include/caffe/layers/python_layer.hpp after merging.

3）

make clean
make -j8 && make pycaffe

基本上我编译的过程中就遇到了这些问题。

编译完成之后，我们现在开始运行一个demo.py程序。在运行之前我们要先下载数据集到本地，总共文件大小是700多MB，可以执行

cd py-faster-rcnn
./data/scripts/fetch_faster_rcnn_models.sh

也可以直接在chrome浏览器输入fetch_faster_rcnn_models.sh文件中数据集的下载地址：

https://dl.dropboxusercontent.com/s/o6ii098bu51d139/faster_rcnn_models.tgz?dl=0

CHECKSUM=ac116844f66aefe29587214272054668

个人觉得使用后面的方法比较快。下载好之后，将文件放在py-faster-rcnn/data文件夹下即可。

然后进入demo.py文件所在位置，执行python文件

cd py-faster-rcnn/tools
python demo.py

然后执行过程中，编译器会不断告诉你缺少哪些包，通过pip进行相应的安装即可。

Pengfei_Chen

关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
Faster R-CNN 安装以及demo的测试

今天开始写下我的第一篇CSDN博客，关于在ubantu上安装Faster R-CNN以及一些编译问题的解决方式，最后通过运行测试源码中的demo.py程序来验证安装成功。在下载源码之前，先看一下自己电脑的硬件配置，我用的电脑配置如下，简单罗列一下CPU，内存以及GPU的信息。model name : Intel(R) Xeon(R) CPU E5-263...
复制链接

扫一扫