多个ImageDataGenerator

Question

我正在尝试从ImageDataGenerator生成两个参数，以便输入到我的model.fit_generator()中，但这不起作用，我现在不知道这样做是否是最好的方法。

我的结构是：

enter image description here

input_imgen1 = ImageDataGenerator(rescale = 1./255, 
                                  vertical_flip=True, 
                                  validation_split=0.2,
                                  horizontal_flip = True)

input_imgen2 = ImageDataGenerator(rescale = 1./255, 
                                  shear_range = 0.2, 
                                  zoom_range = 0.2,
                                  rotation_range=5.)

testgenerator = ImageDataGenerator(rescale = 1./255)

def generate_generator_multiple(generator1, generator2, train_data_dir, batch_size, img_height, 
img_width):
genX1 = generator1.flow_from_directory(train_data_dir,
                                      target_size = (img_height, img_width),
                                      class_mode = 'categorical',
                                      batch_size = batch_size,
                                      shuffle=False, 
                                      seed=7)

genX2 = generator2.flow_from_directory(train_data_dir,
                                      target_size = (img_height, img_width),
                                      class_mode = 'categorical',
                                      batch_size = batch_size,
                                      shuffle=False, 
                                      seed=7)
while True:
        X1i = genX1.next()
        X2i = genX2.next()
        yield [X1i[0], X2i[0]], X2i[1]  #Yield both images and their mutual label


data_gen_train=generate_generator_multiple(generator1=input_imgen1,
                                           generator2=input_imgen2,
                                           train_data_dir=train_dir,
                                           batch_size=batch_size,
                                           img_height=IMG_HEIGHT,
                                           img_width=IMG_WIDTH)    
history = model.fit_generator(
    data_gen_train,
    epochs=epochs,
    steps_per_epoch=25,
    verbose=1,
    validation_data=testgenerator,
    validation_steps=25,
    callbacks=[checkpoint, early_stop, tensor_board]
)

当我适合时出错：

enter image description here

Answer 1

从日志中可以明显看出，您的错误是在验证过程中导致的，应该以与data_gen_train相同的方式构建data_gen_valid。

因此，如果你的训练数据是两个生成器的串联，那么你的验证数据也应该是。