python - 在 python 中读取 *.mhd/*.raw 格式

Question

谁能告诉我如何在 python 中读取包含 .mhd/.raw 文件的数据集？

Answer 1

最简单的方法是使用 SimpleITK (MedPy 也将 ITK 用于 .mhd/.raw 文件)。命令

pip install SimpleITK

适用于许多 python 版本。要阅读 .mhd/.raw，您可以使用此代码 from kaggle

import SimpleITK as sitk
import numpy as np
'''
This funciton reads a '.mhd' file using SimpleITK and return the image array, origin and spacing of the image.
'''

def load_itk(filename):
    # Reads the image using SimpleITK
    itkimage = sitk.ReadImage(filename)

    # Convert the image to a  numpy array first and then shuffle the dimensions to get axis in the order z,y,x
    ct_scan = sitk.GetArrayFromImage(itkimage)

    # Read the origin of the ct_scan, will be used to convert the coordinates from world to voxel and vice versa.
    origin = np.array(list(reversed(itkimage.GetOrigin())))

    # Read the spacing along each dimension
    spacing = np.array(list(reversed(itkimage.GetSpacing())))

    return ct_scan, origin, spacing