为什么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)