在Keras ImageDataGenerator流程方法中遇到save_to_dir問題

Question

我想保存ImageDataGenerator創建的增強圖像，以便稍後使用它們。當我執行下面的代碼，它運行良好，但我希望將圖像保存不我想將它們保存在目錄中出現。

gen = image.ImageDataGenerator(rotation_range=17, width_shift_range=0.12, 
        height_shift_range=0.12, zoom_range=0.12, horizontal_flip=True, dim_ordering='th') 

batches = gen.flow_from_directory(path+'train', target_size=(224,224), 
     class_mode='categorical', shuffle=False, batch_size=batch_size, save_to_dir=path+'augmented', save_prefix='hi') 

我覺得我不能使用此功能正確。任何想法我做錯了什麼？

Answer 1

gen.flow_from_directory給你一個發電機。圖像並不是真正生成的。爲了獲得圖像，你可以迭代發生器。例如

i = 0 
for batch in gen.flow_from_directory(path+'train', target_size=(224,224), 
    class_mode='categorical', shuffle=False, batch_size=batch_size, 
    save_to_dir=path+'augmented', save_prefix='hi'): 

    i += 1 
    if i > 20: # save 20 images 
     break # otherwise the generator would loop indefinitely