首先要安装了torchio,安装指令:
pip install torchio
torchio:一个 Python 库,用于在深度学习中高效加载、预处理、增强和基于补丁的医学图像采样。
其次是找到.nii数据的相对位置或者绝对位置,修改到自己代码的相应位置。代码如下:
import torchio as tio flair_path = './data/MICCAI_BraTS2020_TrainingData/BraTS20_Training_003/BraTS20_Training_003_flair.nii' t1_path = './data/MICCAI_BraTS2020_TrainingData/BraTS20_Training_003/BraTS20_Training_003_t1.nii' t1ce_path = './data/MICCAI_BraTS2020_TrainingData/BraTS20_Training_003/BraTS20_Training_003_t1ce.nii' t2_path = './data/MICCAI_BraTS2020_TrainingData/BraTS20_Training_003/BraTS20_Training_003_t2.nii' seg_path = './data/MICCAI_BraTS2020_TrainingData/BraTS20_Training_003/BraTS20_Training_003_seg.nii' flair_img = tio.ScalarImage(flair_path) t1_img = tio.ScalarImage(t1_path) t1ce_img = tio.ScalarImage(t1ce_path) t2_img = tio.ScalarImage(t2_path) seg_img = tio.LabelMap(seg_path) flair_img.plot() t1_img.plot() t1ce_img.plot() t2_img.plot() seg_img.plot()
显示结果:



