tf-faster-rcnn GPU配置详细步骤

Environment

Ubuntu 16.04
NVIDIA GTX 1060
CUDA 9.0
CuDNN 7.1.4
python3

Installation

• Install opencv, cython, easydict

pip install opencv
pip install cython
pip install easydict

• Clone the repository

git clone https://github.com/endernewton/tf-faster-rcnn.git

• Build the cython modules

cd tf-faster-rcnn/lib
vim Makefile
#line 2：python -> python3

make clean
make
cd ..

• Install the Python COCO API

cd data
git clone https://github.com/pdollar/coco.git
cd coco/PythonAPI

vim Makefile
#line 2：python -> python3

make
cd ../..

• Setup data(e.g: PASCAL VOC 2007)

wget http://host.robots.ox.ac.uk/pascal/VOC/voc2007/VOCtrainval_06-Nov-2007.tar
wget http://host.robots.ox.a