【目标检测二】 Faster R-CNN训练、测试自己的数据

有多个Tensorflow版本的faster r-cnn。本文用的版本如下：

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

不同版本实现代码上有小差异。

使用流程图如下。

说明

1.搭建环境：
Tensorflow-gpu==1.2.1
Cuda8.0
Cudnn5.1

2.在命令行输入以下

  ```

./data/scripts/fetch_faster_rcnn_models.sh 

NET=res101
  TRAIN_IMDB=voc_2007_trainval+voc_2012_trainval
  mkdir -p output/${NET}/${TRAIN_IMDB}
  cd output/${NET}/${TRAIN_IMDB}
  ln -s ../../../data/voc_2007_trainval+voc_2012_trainval ./default
  cd ../../..
  ```

这个脚本提供的路径好像下不了了。去readme里面找个路径。

 

3.制作数据，放到data里面

4.pascal_voc里面修改自己数据的类别

5./experiments/script/里面的训练脚本和测试脚本可以改迭代次数，学习率什么的。更多的参数，在config.py里面改。

6.测试的时候用的是demo.py。先修改以下类别再用。

关于demo.py批量测试，参考以下。

https://blog.csdn.net/gusui7202/article/details/83239142

https://blog.csdn.net/gusui7202/article/details/83240212

https://blog.csdn.net/gusui7202/article/details/83412943