医学图像nii格式文件和pkl文件互相转化

 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)

请问这个该如何解决啊?

  • 8
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值