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之間進行數據交換和操作,從而實現更復雜的圖像處理任務。