keras的数据生成器中生成的图像和掩码不匹配

Question

使用Keras中的数据生成器，我尝试将增强的图像和掩码写入磁盘。我正在执行以下操作，但生成的增强蒙版与生成的增强图像不匹配：

image_datagen.fit(images, augment=True, seed=seed) mask_datagen.fit(masks, augment=True, seed=seed)

image_gen=image_datagen.flow(images, batch_size=1,shuffle=False, save_to_dir=dir_image, save_prefix='aug',save_format='png')

mask_gen=image_datagen.flow(masks, batch_size=1,shuffle=False, save_to_dir=dir_mask, save_prefix='aug',save_format='png')

i = 0 
for batch in zip(image_gen, mask_gen):
    i += 1 
    if i > 10: break

Answer 1

我自己发现了问题所在。我应该使用相同的种子:image_gen=image_datagen.flow( seed=seed，batch_size=1，shuffle=False，save_to_dir=dir_image，save_prefix='aug'，save_format='png')

Mask_gen=image_datagen.flow(掩码，seed=seed，batch_size=1，shuffle=False，save_to_dir=dir_mask，save_prefix='aug'，save_format='png')