mask rcnn 訓練自己的數據踩坑記

1.安裝gpu環境:①cuda9,②cudnn7.0.5,粘貼到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0,③conda create -n tensorflow-gpu python=3.5,④一定要pip install tensorflow-gpu==1.5.1;pip install keras==2.0.2,差什麼包再裝。

2.用labelme3.2,注意json_to_dataset的格式,label.png是十六位的,info.yaml是這樣的

3.訓練時,cpu可以運行,gpu反而不行,是圖片過大,更改

4.測試時,model_path爲空,該行改爲

原圖是灰度圖像,增加語句image = skimage.color.gray2rgb(image);保存測試結果:https://blog.csdn.net/yql_617540298/article/details/81123147

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章