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

 

以上。

 

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