Есть ли в SimpleITK трансляция? - PullRequest
Новая
       14

Есть ли в SimpleITK трансляция?

0 голосов
/ 06 февраля 2019

Если я хочу умножить трехмерную плоскость объема на плоскость с 2D-срезом в numpy, я могу использовать трансляцию: 

import numpy as np
vol = np.random.rand(10, 20, 30)
slc = np.random.rand(10, 30)
new_vol = vol * slc[:, None]

Если я попробую что-то подобное в SimpleITK, я получу ошибку 

import SimpleITK as sitk
vol_img = sitk.GetImageFromArray(vol)
slc_img = sitk.GetImageFromArray(slc[:, None])
new_vol_img = vol_img * slc_img

---------------------------------------------------------------------------
RuntimeError                              Traceback (most recent call last)
<ipython-input-1-7d2c0160b591> in <module>
      9 vol_img = sitk.GetImageFromArray(vol)
     10 slc_img = sitk.GetImageFromArray(slc[:, None])
---> 11 new_vol_img = vol_img * slc_img

~\AppData\Local\Continuum\anaconda3\lib\site-packages\SimpleITK\SimpleITK.py in __mul__(self, other)
   4273     def __mul__( self, other ):
   4274         if isinstance( other, Image ):
-> 4275            return Multiply( self, other )
   4276         try:
   4277            return Multiply( self, float(other) )

~\AppData\Local\Continuum\anaconda3\lib\site-packages\SimpleITK\SimpleITK.py in Multiply(*args)
  50874
  50875     """
> 50876     return _SimpleITK.Multiply(*args)
  50877 class N4BiasFieldCorrectionImageFilter(ImageFilter_0):
  50878     """

RuntimeError: Exception thrown in SimpleITK Multiply: C:\b\3.6-64\ITK\Modules\Core\Common\src\itkDataObject.cxx:393:
Requested region is (at least partially) outside the largest possible region.

1 Ответ

0 голосов
/ 07 февраля 2019

Это нельзя сделать напрямую в SimpleITK, поскольку концепция изображения не эквивалентна массиву интенсивностей, она имеет физический пространственный экстент (см. это объяснение для чтения документов ).Интервал, начало и направление двух умноженных изображений должны быть одинаковыми.

Чтобы сделать то, что вы хотите, вам нужно будет выполнить итерации по фрагментам, а затем заново составить том.

ЗдесьПример кода, который делает именно это: 

import SimpleITK as sitk

img = sitk.ReadImage('training_001_ct.mha')
slc = sitk.GridSource(outputPixelType=img.GetPixelID(), size=img.GetSize()[0:2], 
                             sigma=(0.1,0.1), gridSpacing=(20.0,20.0))
slc.SetSpacing(img.GetSpacing()[0:2])

modified_slices = []
for i in range(img.GetDepth()):
    current_img_slc = img[:,:,i]
    slc.SetOrigin(current_img_slc.GetOrigin())
    slc.SetDirection(current_img_slc.GetDirection())
    modified_slices.append(current_img_slc*slc)

sitk.Show(sitk.JoinSeries(modified_slices))

Пожалуйста, опубликуйте будущие вопросы на форуме ITK Discourse и используйте тег simpleitk.

...