获取文件夹下文件名/读取放缩图片/存储数据

导入文件夹内的图片数据

# 获取文件名
import os

all_file = []
def file_name(path): 
    for file_name in os.listdir(path):
        all_file.append(file_name)

path = 'test/'
file_name(path)
num_image_train = len(all_file)

# 导入训练数据
x_train = np.zeros((num_image_train, 224, 224, 3))

for i in range(num_image_train):
    img = Image.open(path + all_file[i])
    img_resize = img.resize((224, 224))
    x_train[i] = np.array(img_resize)

np.save("x_train.npy", x_train)
print("训练数据导入完毕")