【目標檢測二】 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