monai如何读取.nii/.nii.gz数据

  1. 单个读取:

    from monai.transforms import LoadImage
    
    img_loader = LoadImage()
    img_array = img_loader("xxx.nii.gz")
    # 读取到的img_array是继承了pytorch tensor的类,可以使用tensor有关的方法
    
  2. 批量读取:

    from monai.transforms import LoadImageD
    
    dict_loader = LoadImageD(keys=("image", "label"))
    data_dict = dict_loader({"image": "image.nii.gz", 
                             "label": "label.nii.gz"})
    # data_dict['image']和data_dict['label']访问读取到的图像数据即可
    

你可能感兴趣的:(小知识点,学习笔记,python,深度学习,pytorch,monai)