简体   繁体   English

Keras:使用flow_from_directory为fit_generator输入多个输入

[英]Keras: Multiple Inputs for fit_generator using flow_from_directory

Is it possible to have two flow_from_directory in a fit_generator? 是否可以在fit_generator中有两个flow_from_directory?

For example 例如

train_generator_1 = train_datagen.flow_from_directory(
    train_data_dir_1,
    target_size=(img_height, img_width),
    batch_size=batch_size,
    class_mode='binary')

train_generator_2 = train_datagen.flow_from_directory(
    train_data_dir_2,
    target_size=(img_height, img_width),
    batch_size=batch_size,
    class_mode='binary')

custom_vgg_model.fit_generator(
    [train_generator_1, train_generator_2],
    steps_per_epoch=nb_train_samples,
    epochs=epochs,
    validation_data=validation_generator,
    validation_steps=nb_validation_samples,
    callbacks=[checkpointer, csv_logger])

You can try making a generator that contains both: 您可以尝试制作包含以下两者的生成器:

In python 3: 在python 3中:

def doubleGenerator(generator1,generator2):

    while True:
        for (x1,y1),(x2,y2) in zip(generator1,generator2):
            yield ([x1,x2],y1)

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM