如何使用python3将多个png或Jpeg图像转换为一个nifti图像？

import nibabel as nib
import numpy as np
from PIL import Image
import os

def load_and_convert_images(image_paths):
    images = []
    for path in image_paths:
        img = Image.open(path)
        img = img.convert('L')  # 将图像转换为灰度图像
        img = np.array(img)  # 将图像转换为NumPy数组
        images.append(img)
    return np.array(images)

image_folder = '/path/to/images'  # 图像文件夹路径
image_paths = [os.path.join(image_folder, f) for f in os.listdir(image_folder) if f.endswith('.png') or f.endswith('.jpeg')]

image_data = load_and_convert_images(image_paths)

nifti_image = nib.Nifti1Image(image_data, np.eye(4))  # 创建NIfTI图像对象
nifti_image.to_filename('/path/to/output.nii.gz')  # 保存为NIfTI文件