为什么Python ITK PyBuffer不接受我的numpy数组？

Question

我正在使用python 2.6和ITK包装器(来自PythonXY 2.6.6.2).我正在尝试将numpy/scipy的3D图像发送到itk进行处理.

import itk
imageType = itk.Image.F3
buf =  scipy.zeros( (100,100,100), dtype = float) 
itkImage = itk.PyBuffer[imageType].GetImageFromArray(buf)

GetImageFromArray()失败,出现以下错误:

RuntimeError: Contiguous array couldn't be created from input python object

但是,如果我自己不创建缓冲区,但让ITK创建图像,GetImageFromArray()突然工作:

import itk
imageType = itk.Image.F3
itkImage1 = imageType.New(Regions=[256, 256, 256])
buf = itk.PyBuffer[imageType].GetArrayFromImage(itkImage1)
itkImage2 = itk.PyBuffer[imageType].GetImageFromArray(buf)

我如何自己创建一个numpy数组,GetImageFromArray()会接受它？

Answer 1

答案很简单:

• 在python中,"float"可能是64位(c中为double).
• 在itk中,F3是一个32位浮点数.

为ndarray指定正确的类型使其工作:

import itk
imageType = itk.Image.F3
buf =  scipy.zeros( (100,100,100), dtype = numpy.float32)
itkImage = itk.PyBuffer[imageType].GetImageFromArray(buf)