溫馨提示×

NumPy怎么與SimpleITK集成使用

小億
105
2024-05-13 14:52:23
欄目: 編程語言

NumPy與SimpleITK可以很方便地進行集成使用，因為SimpleITK的圖像數(shù)據(jù)類型與NumPy數(shù)組之間可以相互轉(zhuǎn)換。以下是一個簡單的示例代碼，演示了如何將SimpleITK圖像轉(zhuǎn)換為NumPy數(shù)組，以及將NumPy數(shù)組轉(zhuǎn)換為SimpleITK圖像：

import SimpleITK as sitk
import numpy as np

# 從文件加載SimpleITK圖像
sitk_image = sitk.ReadImage("image.nii")

# 將SimpleITK圖像轉(zhuǎn)換為NumPy數(shù)組
numpy_array = sitk.GetArrayFromImage(sitk_image)

# 將NumPy數(shù)組轉(zhuǎn)換為SimpleITK圖像
sitk_image_new = sitk.GetImageFromArray(numpy_array)

# 保存SimpleITK圖像到文件
sitk.WriteImage(sitk_image_new, "output_image.nii")

通過這種方式，可以方便地在NumPy和SimpleITK之間進行數(shù)據(jù)交換和操作，從而實現(xiàn)更復(fù)雜的圖像處理任務(wù)。

0 踩