nii文件转pkl,根据以下代码可以完成。
'''
nii转pkl
'''
# 读取.nii文件
nii_file = 'data/LPBA40/test/S01.delineation.skullstripped.nii.gz'
img = nib.load(nii_file)
# 获取图像数据
img_data = img.get_fdata()
img_affine = img.affine # 获取affine矩阵
# 序列化到.pkl文件
with open('data/LPBA40/output.pkl', 'wb') as f:
pickle.dump(img_data, f)
pkl转为nii格式,报错。
''
pkl转nii
'''
with open('data/LPBA40/output.pkl', 'rb') as f:
loaded_data = pickle.load(f)
print(loaded_data) # 查看加载的数据结构
# 从.pkl文件反序列化图像数据和affine矩阵
with open('data/LPBA40/output.pkl', 'rb') as f:
img_data, affine = pickle.load(f) # 同时加载图像数据和affine矩阵
# 创建一个新的Nifti1Image,使用加载的affine矩阵
new_img = nib.Nifti1Image(img_data, affine=affine)
# 保存为.nii文件
new_img_file = 'data/LPBA40/restored_image.nii.gz'
new_img.to_filename(new_img_file)
报错:
img_data, affine = pickle.load(f) # 同时加载图像数据和affine矩阵
ValueError: too many values to unpack (expected 2)
请问这个该如何解决啊?