2017-09-27 33 views
-1

fit_generatorに2つのflow_from_directoryを含めることはできますか?例Keras:flow_from_directoryを使用するfit_generatorの複数の入力

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]) 
+1

次の2つの並列入力をしたいですか?あなたのモデルは2つの入力を使用していますか?出力はどうですか? 2つのパラレル出力? –

+0

@DanielMöllerはい、私のモデルは2つのCNNを持っています。それぞれが異なる入力を持つ。出力は同じです。入力は基本的に同じ画像ですが、各CNNは異なる前処理画像を持っています。 –

+0

Ok、出力...ジェネレータ1またはジェネレータ2から取っておきたいですか? –

答えて

1

については

あなたは、両方が含まれている発電機を作ってみることができます。

のpython 3では:

def doubleGenerator(generator1,generator2): 

    while True: 
     for (x1,y1),(x2,y2) in zip(generator1,generator2): 
      yield ([x1,x2],y1) 
関連する問題