python - 读取图像文件

Question

我有一个包含 4 个文件夹（1、2、3、4）的目录。每个文件夹中都有 jpg 图像。我使用下面的代码来读取图像。问题是所有图像的形状都不同。所以，现在我有一个图像列表，每个图像都有不同的形状。

1）有没有更好的方法从目录中读取 img 文件？（也许直接分配给一个 numpy 数组）

2）如何调整图像大小以使它们都具有相同的形状？

谢谢！

import imageio
import os.path

images = []

for folder in os.listdir('images'):
    for filename in os.listdir('images/'+folder):
        if filename.endswith(".jpg"): 
            img = imageio.imread('images/'+folder+'/'+filename)
            img.reshape((1,img.flatten().shape[0])).shape
            images.append(img)

标签： pythonpython-3.xpython-imageio