Импорт нескольких изображений

Question

Я хочу дублировать мои фотографии с этим кодом:

from keras.preprocessing.image import ImageDataGenerator, array_to_img, img_to_array, load_img

datagen = ImageDataGenerator(
        rotation_range=40,
        width_shift_range=0.2,
        height_shift_range=0.2,
        shear_range=0.2,
        zoom_range=0.2,
        horizontal_flip=True,
        fill_mode='nearest')

img = load_img('data/class1/11408_3.jpg')
x = img_to_array(img)  
x = x.reshape((1,) + x.shape)

i = 0
for batch in datagen.flow(x, batch_size=1,
                          save_to_dir='preview', save_prefix='class1', save_format='jpeg'):
    i += 1
    if i > 20:
        break 

С одним изображением все работает отлично.Но у меня много картинок на этом пути.Как я могу получить все фотографии - одну за другой?

Answer 1

Спасибо за ваш вклад.Я теперь решил это с:

from keras.preprocessing.image import ImageDataGenerator, array_to_img, img_to_array, load_img
import glob, os

datagen = ImageDataGenerator(
        rotation_range=90,
        horizontal_flip=True,
        fill_mode='nearest')

os.chdir("dir")
for file in glob.glob("*.png"):
    img = load_img(file)
    x = img_to_array(img)
    x = x.reshape((1,) + x.shape)

    i = 0
    for batch in datagen.flow(x, batch_size=1,save_to_dir='output', save_prefix='new', save_format='jpg'):
        i += 1
        if i > 20:
            break 

Answer 2

Вы можете использовать список строк, который содержит каждый путь к файлу, и поместить его в цикл for следующим образом:

filePaths = ['data/class1/11408_3.jpg', ...]
for file in filePaths:
    img = load_img(file)