Keras - 如何在不变形宽高比的情况下使用ImageDataGenerator

Question

我有一个包含不同大小图像的文件夹.我尝试将带有flow_from_directory的ImageDataGenerator用于批量加载/数据扩充.

有没有办法保持我的图像的纵横比？似乎图像被拉伸到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上打开了拉取请求以添加此功能.