TensorFlow deeplab V3+训练自己的数据踩坑记

作为一个小白，制作好了训练数据，代码也都按照指示修改好了，准备train的时候报了一些错误，和大家一起分享踩坑的解决方法。

我的TensorFlow版本是1.10

1.

ValueError: The specified dataset is not supported yet.

这是在提示自己做的训练样本没有注册，虽然前期按照一些博主分享的教程修改了segmentation_dataset.py文件，但是仍然提示没有注册。

解决方法：修改…/research/deeplab/datasets/data_generator.py，在与segmentation_dataset.py中相同的位置注册自己的数据集。

参考https://blog.csdn.net/xjtdw/article/details/92848032

2.

crop_width=self.crop_size[1],  IndexError: list index out of range 1

给train.py赋值出现的问题。

解决方法：–train_crop_size=512 –train_crop_size=512 修改为–train_crop_size=‘513,513’ （此处crop大小可根据自己的数据来定）

参考https://blog.csdn.net/xjtdw/article/details/92848032

3.

AttributeError: module 'tensorflow.image' has no attribute 'resize'

可能是我的tf版本比较旧，接口不对。

解决方法：找到报错的py文件对应位置，修改为 tf.image.resize_images

4.

AttributeError: module 'tensorflow.math' has no attribute 'divide'

产生问题的原因和3差不多。

解决方法：修改对应位置的调用代码为 tf.divide

参考：https://blog.csdn.net/oMoDao1/article/details/82284105

 

以上。