这个问题干扰我一天的进度了。
目前,终于找到!!!!
没有报错的情况下,大概率是导入数据问题。我先说下我的具体的check步骤。
最后发现os.listdir() 在ubuntu中是乱序读入数据的。。。所以模型data与label压根没配对。。。
以此为戒,希望后续牢记。关于,集群debug,参考博客。
我最后的处理方法。(我的data是png,label是tiff)
self.data_name = os.listdir(data_path)
# self.label_name = os.listdir(label_path)
self.label_name = self.data_name.copy()
self.label_name = [name.replace(".png", ".tiff") for name in self.label_name]
self.data_path = []
self.label_path = []
for data, label in zip(self.data_name, self.label_name):
self.data_path.append(data_path + "/" + data)
self.label_path.append(label_path + "/" + label)