我有一个包含 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)