Faster rcnn 训练自己的数据—环境搭建

faster-rcnn Python版本源码地址：https://github.com/rbgirshick/py-faster-rcnn

这篇文章主要介绍搭建用faster-rcnn进行目标检测所需的环境。

1.电脑上已经有可运行caffe所需的环境

2.下载faster-rcnn python版本源码

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

# In your Makefile.config, make sure to have this line uncommented  
WITH_PYTHON_LAYER := 1  
# Unrelatedly, it's also recommended that you use CUDNN  
USE_CUDNN := 1

4. 然后在终端 make pycaffe 一下，一定要执行这条指令，不然后面会报错：No import named _caffe.

5. 运行faster-rcnn里的demo.py

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

出现下图表示，环境已经搭建好了。