simpleitk_plugin.py 531 B

1234567891011121314151617181920212223
  1. __all__ = ['imread', 'imsave']
  2. try:
  3. import SimpleITK as sitk
  4. except ImportError:
  5. raise ImportError(
  6. "SimpleITK could not be found. "
  7. "Please try "
  8. " easy_install SimpleITK "
  9. "or refer to "
  10. " http://simpleitk.org/ "
  11. "for further instructions."
  12. )
  13. def imread(fname):
  14. sitk_img = sitk.ReadImage(fname)
  15. return sitk.GetArrayFromImage(sitk_img)
  16. def imsave(fname, arr):
  17. sitk_img = sitk.GetImageFromArray(arr, isVector=True)
  18. sitk.WriteImage(sitk_img, fname)