python - Keras-如何使用ImageDataGenerator而不改变宽高比

我有一个包含不同尺寸图像的文件夹。我试图将ImageDataGenerator与flow_from_directory一起使用以进行批量加载/数据扩充。

有没有办法保持图像的长宽比？似乎图像已被拉伸(stretch)到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)

图像被拉伸(stretch)到(256,256)。

最佳答案

我找到了问题的答案。

目前，在使用ImageDataGenerator/flow_from_directory时，无法保持纵横比，但是在Github上打开了pull request来添加此功能。

关于python - Keras-如何使用ImageDataGenerator而不改变宽高比，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/42467734/