使用
zip()合并生成器会导致生成无限迭代器。使用此代替:
def combine_generator(gen1, gen2): while True: yield(next(gen1), next(gen2))
修改后的代码如下所示:
datagen_args = dict(rotation_range=10, width_shift_range=0.1, height_shift_range=0.1, horizontal_flip=True)in_gen1 = ImageDataGenerator(**datagen_args)in_gen2 = ImageDataGenerator(**datagen_args)def combine_generator(gen1, gen2): while True: yield(next(gen1), next(gen2))train_generator = combine_generator(in_gen1, in_gen2)model.fit(train_generator.flow([pair_df[:, 0,::],pair_df[:, 1,::]], y_train,batch_size=16), epochs, verbose = 1)
请参阅此线程以获取更多参考。



