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