通过nibabel实现nii.gz转npy

nii_2_npy

import nibabel as nib
import os
import numpy as np

img_path = r"E:My_jobCAT08train_datadataset01/"
# seg_path = "/home/lei/train/seg/"
saveimg_path = r"medimage/"
# saveseg_path = "/home/lei/train/npy_seg/"

img_names = os.listdir(img_path)
# seg_names = os.listdir(seg_path)

for img_name in img_names:
    if "nii" in img_name:
        print(img_name)
        img = nib.load(img_path + img_name).get_data()  # 载入
        img = np.array(img)
        np.save(saveimg_path + str(img_name).split(".")[0] + ".npy", img)  # 保存

# for seg_name in seg_names:
#     print(seg_name)
#     seg = nib.load(seg_path + seg_name).get_data()
#     seg = np.array(seg)
#     np.save(saveseg_path + str(seg_name).split(".")[0] + ".npy", img)  # 保存