Keras seed 不会增强图像并掩盖相同的内容

Question

我正在尝试同步转换我的图像和 mask 标签，但随机增强以不同方式应用于两个生成器(当我将两个图像保存到一个目录时证明)。

我试过了

• 在我的库堆栈中播种所有生成器
• 禁用随机播放和多处理
• 确保图像和蒙版之间的增强选项相同

将批量大小设置为 1 确实有效，但随后我的训练性能受到影响。

seed_val=0

data_gen_args = dict(
    rescale=1./255,
    horizontal_flip=True,
    validation_split=validation_split)

image_datagen = ImageDataGenerator(**data_gen_args)
mask_datagen = ImageDataGenerator(**data_gen_args)

image_generator = image_datagen.flow_from_directory(
    'data/x/train_images/',
    target_size=(224, 224),
    color_mode='rgb',
    class_mode=None,
    batch_size=batch_size,
    subset='training',
    save_to_dir='tmp/img/',
    seed=seed_val)

mask_generator = mask_datagen.flow_from_directory(
    'data/x/train_annotations/',
    target_size=(224, 224),
    color_mode='grayscale',
    class_mode=None,
    batch_size=batch_size,
    subset='training',
    save_to_dir='tmp/mask/',
    seed=seed_val)

train_generator = zip((image_generator), (mask_generator))

Answer 1

尝试将图像和蒙版的 shuffle 设置为 False

image_generator = image_datagen.flow_from_directory(
    'data/x/train_images/',
    target_size=(224, 224),
    shuffle = False,
    color_mode='rgb',
    class_mode=None,
    batch_size=batch_size,
    subset='training',
    save_to_dir='tmp/img/',
    seed=seed_val)

mask_generator = mask_datagen.flow_from_directory(
    'data/x/train_annotations/',
    target_size=(224, 224),
    shuffle = False,
    color_mode='grayscale',
    class_mode=None,
    batch_size=batch_size,
    subset='training',
    save_to_dir='tmp/mask/',
    seed=seed_val)