nii.gz格式四维转三维（给爷写笑了）

把这里当一下垃圾篓(别管我啦 发疯.jpg

import nibabel as nib

def convert_4d_to_3d(input_file, output_file, drop_axis=2):
    # 加载四维的nii.gz文件
    img = nib.load(input_file)

    # 获取四维数据
    data = img.get_fdata()

    # 丢弃指定的维度，这里假设drop_axis为3表示丢弃第四个维度（即时间维度）
    # 如果drop_axis为0，则丢弃第一个维度，以此类推
    new_data = data.squeeze(axis=drop_axis)

    # 创建一个新的affine矩阵，因为丢弃维度后空间坐标可能会改变
    # 这里我们简单地复制原始affine，但根据具体情况可能需要调整
    new_affine = img.affine.copy()

    # 创建新的header，并更新维度信息
    header = img.header.copy()
    header.set_data_shape(new_data.shape)

    # 使用新的数据和header创建一个新的nibabel图像对象
    new_img = nib.Nifti1Image(new_data, new_affine, header)

    # 保存为新的三维nii.gz文件
    nib.save(new_img, output_file)

input_file = 'E:\\zxxzl\\srtp02\\four_jiqixuexi\\pngtransform\\pngto3d_test\\output\\1_4d.nii.gz'  # 替换为您的四维nii.gz文件的路径
output_file = 'E:\\zxxzl\\srtp02\\four_jiqixuexi\\pngtransform\\pngto3d_test\\output\\1_4d_3d.nii.gz'  # 您想要保存的三维nii.gz文件的路径
convert_4d_to_3d(input_file, output_file)

output_path = 'E:\\zxxzl\\srtp02\\four_jiqixuexi\\pngtransform\\pngto3d_test\\output\\1_4d_3d.nii.gz'
# 加载保存的NIfTI文件以验证维度
loaded_nii = nib.load(output_path)
print("Loaded image dimensions:", loaded_nii.shape)
print("Loaded image affine:", loaded_nii.affine)

批量四维转三维

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

# 读取PNG文件夹路径
png_folder = "E:/zxxzl/srtp02/four_jiqixuexi/pngtransform/nii4dto3d/4ddeck"
# 创建保存NIfTI文件的目标文件夹
output_folder = "E:/zxxzl/srtp02/four_jiqixuexi/pngtransform/nii4dto3d/output"
os.makedirs(output_folder, exist_ok=True)

def convert_4d_to_3d(input_file, output_file, drop_axis=2):
    # 加载四维的nii.gz文件
    img = nib.load(input_file)

    # 获取四维数据
    data = img.get_fdata()

    # 丢弃指定的维度，这里假设drop_axis为3表示丢弃第四个维度（即时间维度）
    # 如果drop_axis为0，则丢弃第一个维度，以此类推
    new_data = data.squeeze(axis=drop_axis)

    # 创建一个新的affine矩阵，因为丢弃维度后空间坐标可能会改变
    # 这里我们简单地复制原始affine，但根据具体情况可能需要调整
    new_affine = img.affine.copy()

    # 创建新的header，并更新维度信息
    header = img.header.copy()
    header.set_data_shape(new_data.shape)

    # 使用新的数据和header创建一个新的nibabel图像对象
    new_img = nib.Nifti1Image(new_data, new_affine, header)

    # 保存为新的三维nii.gz文件
    nib.save(new_img, output_file)

for filename in os.listdir(png_folder):
    if filename.endswith(".nii.gz"):
        input_file = os.path.join(png_folder, filename)
        # 构建目标文件的完整路径和名称
        nii_filename = os.path.splitext(filename)[0] + '.gz'
        output_file = os.path.join(output_folder, nii_filename)
        convert_4d_to_3d(input_file, output_file)


output_path1 = "E:/zxxzl/srtp02/four_jiqixuexi/pngtransform/nii4dto3d/output/1_mask.nii.gz"
loaded_nii = nib.load(output_path1)
print("Loaded image dimensions:", loaded_nii.shape)
print("Loaded image affine:", loaded_nii.affine)