小编Yae*_*l N的帖子

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

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

AttributeError: 'zip' object has no attribute 'shape'
Run Code Online (Sandbox Code Playgroud)

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

AttributeError: 'zip' object has no attribute 'shape'
Run Code Online (Sandbox Code Playgroud)

我将它们结合在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)

Run Code Online (Sandbox Code Playgroud)

并将它们输入到fit_generator(...)

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

train_generator = zip(input_gen, output_gen)
Run Code Online (Sandbox Code Playgroud)

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

tensorflow tf.keras

7
推荐指数
1
解决办法
1026
查看次数

标签 统计

tensorflow ×1

tf.keras ×1