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