如何将目录中的图像转换为np数组

import os
import cv2
import numpy as np

def convert_images_to_np_array(directory):
    images = []
    for filename in os.listdir(directory):
        if filename.endswith(".jpg") or filename.endswith(".png"):
            image_path = os.path.join(directory, filename)
            image = cv2.imread(image_path)
            if image is not None:
                images.append(image)
    np_array = np.array(images)
    return np_array

directory = "path/to/directory"
np_array = convert_images_to_np_array(directory)