.nii格式的图像转化numpy格式

主要使用nibabel

import nibabel as nib
import os
import numpy as np

img_path = '/home/lei/train/img/'
seg_path = '/home/lei/train/seg/'
saveimg_path = '/home/lei/train/npy_img/'
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:
    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’, seg)