Faster R-CNN GPU训练运行说明

本篇文章主要是记录本人学习Faster R-CNN的有关内容，这次是记录使用GPU运行模型训练的步骤：

1、首先下载模型的源代码（本人使用的是tensorflow版的代码）：

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

2、配置环境

训练模型需要提前安装好相关的依赖包：

Cython           0.28.2

easydict         1.9

matplotlib       2.2.2

numpy            1.15.4

opencv-python    3.4.4.19

pandas           0.23.0

Pillow           5.3.0

scipy            1.1.0

tensorflow-gpu   1.10.1

3、Build the Cython modules

cd ./lib
make clean
make
cd ..

4、安装COCO API

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

5、下载训练好的vgg16模型

mkdir -p data/imagenet_weights
cd data/imagenet_weights
wget -v http://download.tensorflow.org/models/vgg_1