NumPy与SimpleITK可以很方便地进行集成使用,因为SimpleITK的图像数据类型与NumPy数组之间可以相互转换。以下是一个简单的示例代码,演示了如何将SimpleITK图像转换为NumPy数组,以及将NumPy数组转换为SimpleITK图像:
import SimpleITK as sitk import numpy as np # 从文件加载SimpleITK图像 sitk_image = sitk.ReadImage("image.nii") # 将SimpleITK图像转换为NumPy数组 numpy_array = sitk.GetArrayFromImage(sitk_image) # 将NumPy数组转换为SimpleITK图像 sitk_image_new = sitk.GetImageFromArray(numpy_array) # 保存SimpleITK图像到文件 sitk.WriteImage(sitk_image_new, "output_image.nii")
通过这种方式,可以方便地在NumPy和SimpleITK之间进行数据交换和操作,从而实现更复杂的图像处理任务。