作爲一個小白,製作好了訓練數據,代碼也都按照指示修改好了,準備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
以上。