我创建了两个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)
对于什么是错的任何想法?