Как при использования генератора ImageDataGenerator избежать деформации изображений после загрузки и сохранить соотношение сторон у фотографий

Есть папка с изображениями разных размеров и сторон. При использование ImageDataGenerator с flow_from_directory для пакетной загрузки таким образом:

test_datagen = ImageDataGenerator(rescale=1./ 255)

test_datagen_flow = test_datagen.flow_from_directory(
    args_folder,
    target_size=(150, 150),
    batch_size=16,
    class_mode='binary')

Изображения растягиваются или сжимаются до target_size. Я хотел бы "дополнить" свои изображения, не деформируя их (например заполняя пробелы черным цветом). Есть ли способ сохранить пропорции изображений при загрузке данных пакетом.


Ответы (1 шт):

Автор решения: tomato-magnet-regulato

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

Увидал тут

from PIL import Image
im = Image.open("logo.jpg")
(width, height) = im.size

После передать в target_size = (width, height)

→ Ссылка