mirror of
https://github.com/wassname/scikit-image.git
synced 2026-06-28 03:19:12 +08:00
Matt McCormick
ad23f203da
IO plugin for SimpleITK, http://simpleitk.org/ imread and imsave implemented. Tests based off the PIL tests.
22 lines
587 B
Python
22 lines
587 B
Python
__all__ = ['imread', 'imsave']
|
|
|
|
try:
|
|
import SimpleITK as sitk
|
|
except ImportError:
|
|
raise ImportError("SimpleITK could not be found. "
|
|
"Please try "
|
|
" easy_install SimpleITK "
|
|
"or refer to "
|
|
" http://simpleitk.org/ "
|
|
"for further instructions.")
|
|
|
|
|
|
def imread(fname):
|
|
sitk_img = sitk.ReadImage(fname)
|
|
return sitk.GetArrayFromImage(sitk_img)
|
|
|
|
|
|
def imsave(fname, arr):
|
|
sitk_img = sitk.GetImageFromArray(arr, isVector=True)
|
|
sitk.WriteImage(sitk_img, fname)
|