温馨提示×

NumPy怎么与SimpleITK集成使用

小亿
105
2024-05-13 14:52:23
栏目: 编程语言

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之间进行数据交换和操作,从而实现更复杂的图像处理任务。

0