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