Keras ImageDataGenerator无法与shuffle一起正常工作

时间:2017-08-21 09:45:07

标签: python machine-learning tensorflow keras

我创建了两个ImageDataGenerator对象来同时处理两个图像(一个图像和他的掩码)来训练一个自动编码器/ u-net。为了随机化数据集,我设置了shuffle=True。我从两个文件夹中获取这些图像(一个用于图像,另一个用于蒙版),因此我使用两个flow_from_directory具有相同的种子。然而,在实践中不起作用并且提取的掩模与提取的图像不对应。两个图像(图像和掩码)都具有相同的文件名。

这是我的代码:

data_gen_args = dict(rescale = 1./255,
                     rotation_range = 5,
                    zoom_range = 0.05,
                    horizontal_flip = True)
image_datagen = ImageDataGenerator(**data_gen_args)
mask_datagen = ImageDataGenerator(**data_gen_args)

seed = 1

image_generator = image_datagen.flow_from_directory(
    'data/train_'+str(shape),
    color_mode = 'grayscale',
    class_mode = None,
    batch_size = batch_size,
    shuffle = True,
    seed = seed)

mask_generator = mask_datagen.flow_from_directory(
    'data/train_mask_'+str(shape),
    color_mode = 'grayscale',
    class_mode = None,
    batch_size = batch_size,
    shuffle = True,
    seed = seed)

对于什么是错的任何想法?

0 个答案:

没有答案