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

 

以上。

 

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