python批量读取文件里的图片_Python批量从文件夹中读取图片

导入必要的库

from PIL import Image

import numpy as np

先导入一张图片

X_train = Image.open(r'.\faces\00b34e16b0270e43d2e6591bcadb2eaf-0.jpg')

X_train = X_train.resize((64,64),Image.ANTIALIAS)

X_train = np.array(X_train)

X_train = np.expand_dims(X_train, axis=0)

print(X_train.shape)

用concatenate的方式导入指定数量的图片

for dirname, _, filenames in os.walk(r'.\faces'):

for filename in filenames:

if X_train.shape[0] > 5000:

break

try:

im = Image.open(os.path.join(dirname, filename))

im = im.resize((64,64),Image.ANTIALIAS)

image_array = np.array(im)

image_array = np.expand_dims(image_array, axis=0)

X_train = np.concatenate((X_train, image_array), axis=0)

except:

pass

print(str(X_train.shape[0]))

相关函数说明

X_train = X_train.resize((64,64),Image.ANTIALIAS)

此代码是指将读取的图片尺寸缩放为64x64，使用ANTIALIAS过滤器。

X_train = np.array(X_train)

是指将X_train从PIL.Image.Image类型转化为nd.array类型。

X_train = np.expand_dims(X_train, axis=0)

是指将(64,64,3)的尺寸拓展为(1,64,64,3)。

如果axis=1，则尺寸变为(64,1,64,3)，以此类推。

X_train = np.concatenate((X_train, image_array), axis=0)

是指将新图片添加到X_train 数组下面，这样一来，每循环一次第一维度会加一。

如果axis=1，则每循环一次第二维度会加一，以此类推。