Keras - 如何在不改變高寬比的情況下使用ImageDataGenerator

Question

我有一個包含不同大小圖像的文件夾。我試圖使用ImageDataGenerator和flow_from_directory進行批量加載/數據增強。

有沒有辦法保持我的圖像的縱橫比？好像被拉伸到target_size圖像：我想「墊」我的圖像，而不使它們變形（以恆定值填充間隙）

這裏是我的代碼：

datagen = ImageDataGenerator(
    rescale = 1./255, 
    fill_mode='constant') 

generator = datagen.flow_from_directory(
    'data/images', 
    target_size=(256,256), 
    color_mode = 'grayscale', 
    batch_size=99, 
    class_mode=None, 
    shuffle=False) 

圖像正在拉伸到（256,256）。

Answer 1

我找到了我的問題的答案。

當使用ImageDataGenerator/flow_from_directory時，目前沒有辦法保持寬高比，但在Github上打開pull request以添加此功能。