如何将文件路径中的图像格式化为与python中的mnist.load_data()相同的格式？

import os
import cv2
import numpy as np
from tensorflow.keras.datasets import mnist

def load_images_from_path(path):
    images = []
    labels = []
    for filename in os.listdir(path):
        if filename.endswith(".jpg") or filename.endswith(".png"):
            img = cv2.imread(os.path.join(path, filename), cv2.IMREAD_GRAYSCALE)
            img = cv2.resize(img, (28, 28))
            img = np.reshape(img, (28, 28, 1))
            images.append(img)
            labels.append(int(filename.split("_")[0]))  # 假设文件名格式为"label_XXX.jpg"
    return np.array(images), np.array(labels)

path = "/path/to/images"  # 图像文件夹路径
images, labels = load_images_from_path(path)