如何从Google Colab的"train“文件夹中加载多个文件夹中的多个图像？

import os
import cv2
import numpy as np

train_folder = "/content/train"
image_size = (224, 224)  # 图像尺寸

images = []
labels = []

for folder_name in os.listdir(train_folder):
    folder_path = os.path.join(train_folder, folder_name)
    if os.path.isdir(folder_path):
        label = folder_name  # 文件夹名称作为标签
        for image_name in os.listdir(folder_path):
            image_path = os.path.join(folder_path, image_name)
            image = cv2.imread(image_path)  # 加载图像
            image = cv2.resize(image, image_size)  # 调整图像尺寸
            images.append(image)
            labels.append(label)

images = np.array(images)
labels = np.array(labels)