将nii格式的文件转换为png

import nibabel as nib
import numpy as np
import matplotlib.pyplot as plt
import os

# 读取NIfTI文件
nii_file = 'a1001.nii'
img = nib.load(nii_file)

# 获取影像数据为numpy数组
data = img.get_fdata()

# 创建保存图片的目录
output_dir = 'output_images'
os.makedirs(output_dir, exist_ok=True)

# 遍历每个切片，并保存为图片
for i in range(data.shape[2]):  # 假设切片在Z轴上
    slice_data = data[:, :, i]  # 获取第i个切片
    plt.imshow(slice_data.T, cmap='gray', origin='lower')  # 显示切片，使用灰度色图
    plt.axis('off')  # 关闭坐标轴

    # 保存图片
    output_path = os.path.join(output_dir, f'slice_{i:03d}.png')
    plt.savefig(output_path, bbox_inches='tight', pad_inches=0)
    plt.close()

print(f"Saved {data.shape[2]} slices to the directory '{output_dir}'.")