Faster-RCNN 论文复现（Tensorflow）

 

Faster-RCNN（TensorFlow代码）

1. 环境配置

硬件说明：

显卡：1080ti（之后会有用）

CUDA：cuda9.0，CUDNN：7.0

Ubuntu：16.0

cat /usr/local/cuda/version.txt  （cuda 版本查看指令）

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2  （cudnn 版本查看指令）

创建一个python3.6的TensorFlow环境tf36

conda create -n tf36 python=3.6

因为Tensoflow在Linux上配置需要对应CUDA版本信息，所以不能直接利用pip install tensorflow-gpu进行安装，需要找到对应的版本信息进行安装。传送门看这里！

根据论文README要求安装其他包，包括numpy，matplotlib，Cython，easydict等安装包，根据报错或者requirements文件进行安装即可！

2.一些调试工作

这部分主要根据README文件进行一些调试，细节如下：

1. 更改GPU结构设置

                            

因为之前说明使用的1080ti，所以需要在lib文件中setup.py中进行设置修改，按照上图要求即可。

2. make文件

make clean
make
cd ..

make文件主要是为了设置Cython，这个很重要，在稍后的报错会因为Makefile而报错，这里随带将其debug。报错内容如下：

ModuleNotFoundError: No module named 'utils.cython_bbox'

 这个报错需要更改两处，第一需要将Lib\utils\bbox.pyx更改我cython.bbox.pyx。

第二：更改Makefile

all:
	python setup.py build_ext --inplace
	rm -rf build
clean:
	rm -rf */*.pyc
	rm -rf */*.so

因为使用的是python3所以需要上段代码更改：（将python改为python3）

all:
	python3 setup.py build_ext --inplace
	rm -rf build
clean:
	rm -rf */*.pyc
	rm -rf */*.so

 3.安装Python cocoAPI

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

4.数据集结构设置，按照论文要求设置即可

5.训练

Shell
  ./experiments/scripts/train_faster_rcnn.sh [GPU_ID] [DATASET] [NET]
  # GPU_ID is the GPU you want to test on
  # NET in {vgg16, res50, res101, res152} is the network arch to use
  # DATASET {pascal_voc, pascal_voc_0712, coco} is defined in train_faster_rcnn.sh
  # Examples:
  ./experiments/scripts/train_faster_rcnn.sh 0 pascal_voc vgg16

剩下的就是一些路径的设置了。

整个代码主要是cython文件还有Makefile文件的一些问题，耐心慢慢debug就行。