1. 程式人生 > >Win10-Tensorflow-FasterRcnn訓練環境配置手冊

Win10-Tensorflow-FasterRcnn訓練環境配置手冊

1、安裝顯示卡驅動:根據自己電腦的顯示卡下載適合的顯示卡驅動

2、安裝CUDA8.0

安裝完成後檢視cuda版本: 在命令列輸入nvcc -V

3、安裝cudnn v6.0

將該資料夾下的cuda資料夾下對應的檔案目錄拷貝到第一步安裝的CUDA  檔案目錄

4、安裝Anaconda3 4.2.0 (對應python3.5

備註:環境變數一般會自動新增

5、tensorflow-gpu 版本tensorflow-gpu-1,3,0.0rc0

6、安裝其他依賴的package:可以直接pipinstall ****

①protobuf-3.5.1 

②tensorflow-tensorboard-0.1.8

③html5lib-0.9999999

④markdown>=2.6.8

⑥bleach-1.5.0

⑦升級pip版本至 pip-9.0.1:python-m pip install --upgrade pip

備註:python的一些模組的下載地址:https://pypi.python.org/pypi

6、安裝python的其他依賴cython, python-opencv, easydict

7、下載faster-rcnn: 

https://github.com/dBeker/Faster-RCNN-TensorFlow-Python3.5

解壓後:


①到目錄./data/coco/PythonAPI下,在命令列輸入:pythonsetup.py build_ext –inplace

②到目錄./data/coco/PythonAPI下,在命令列輸入:pythonsetup.py build_ext install

備註:可以下載Pretrain-model的地址: 

https://github.com/tensorflow/models/tree/master/research/slim#pre-trained-models

③訓練資料存放格式:資料存放目錄:data/VOCDevkit2007/VOC2007


Annotations資料夾放XML標註檔案

ImageSets資料夾有個Main資料夾,放test.txt和trainval.txt

JPEGImages資料夾放所有的訓練資料

8、開始訓練:在命令列python train.py

9、配置過程中可能出現出現錯誤:

①numpy版本升級


②opencv與python版本對應問題(原本VS版本是2013, 可能要換成2015)

③在開始訓練時出現錯誤:

ImportError: No module named '_pywrap_tensorlfow_internal

檢查環境變數,cuda\bin沒有在環境變數path裡,將.\CUDA\bin 加入到環境變數path。在裝cuda時 就有提示了。 需要安裝後自己改下路徑(把路徑縮短),再手動加入環境變數