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