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

你可能感兴趣的:(mask rcnn 训练自己的数据踩坑记)