Как сохранить изображения после увеличения данных в новой папке без зацикливания - PullRequest
Новая
       48

Как сохранить изображения после увеличения данных в новой папке без зацикливания

0 голосов
/ 17 октября 2019

Я пытаюсь сохранить мои дополненные изображения в папке. Но цикл выполняет бесконечность раз. У меня в папке 5000 изображений, но количество получаемых увеличенных изображений бесконечно. Моя цель - получить такое же количество дополненных изображений, т. Е. 5000.

Спасибо 

import numpy as np
from keras.preprocessing.image import ImageDataGenerator

datagen = ImageDataGenerator(rotation_range=90)

image_path = 'C:/Users/1/Desktop/DEEP/Dataset/Train/1training_c10882.png'

image = np.expand_dims(imageio.imread(image_path), 0)

save_here = 'D:\Augmented DATASET\'


generator = datagen.flow_from_directory('C:/Users/1/Desktop/DEEP/Dataset/Train',target_size=(224,224),
                                    batch_size = 256, class_mode = 'binary')


for inputs,outputs in generator:
    pass
...