keras fit_generator：'zip'对象没有属性'shape'

Question

像（图像，遮罩）数据生成器中的keras文档中一样，我创建了数据生成器以具有（输入，输出）元组图像。
当在我的笔记本电脑上（使用tensorflow 1.13.1）在本地运行时，它没有问题运行，但是当在服务器上（使用tensorflow- gpu 1.13.1）运行时，出现错误：

AttributeError: 'zip' object has no attribute 'shape'

如示例所示，在使用flow_from_directory创建两个生成器之后：

AttributeError: 'zip' object has no attribute 'shape'

我将它们结合在keras文档中：

from tensorflow import keras

...

input_gen = input_datagen.flow_from_directory(
        directory=os.path.join(path_in, dirname),
        class_mode=None,
        color_mode=color_mode,
        batch_size=batch_size,
        target_size=(img_size, img_size),
        seed=seed_in)

并将它们输入到fit_generator（...）

目前解决它使用一个函数来产生新的联合发电机的建议在这里：

train_generator = zip(input_gen, output_gen)

虽然我仍然想了解为什么tensorflow cpu版本没有出现此错误，以及为什么gpu版本不支持zip ...

Answer 1

看起来 Tensorflow 1.13 中对 fit_generator 的支持已更改。正如本期中提到的解决方法，

代替

train_generator = zip(input_gen, output_gen)

和

train_generator = (pair for pair in zip(input_gen, output_gen))