vtk 3D图像的旋转/平移与插值(python)

时间:2016-05-26 16:37:57

标签: python image-processing interpolation vtk image-rotation

我有2个矩阵:

#for example
rotation = matrix([[ 0.61782155,  0.78631834,  0.        ],
            [ 0.78631834, -0.61782155,  0.        ],
            [ 0.        ,  0.        , -1.        ]])
translation = matrix([[-0.33657291],
            [ 1.04497454],
            [ 0.        ]])
vtkinputpath = "/hello/world/vtkfile.vtk"
vtkoutputpath = "/hello/world/vtkrotatedfile.vtk"
interpolation = "linear"

我有一个包含3D图像的vtk文件,我想在python中创建一个函数来旋转/平移插值。

import vtk

def rotate(vtkinputpath, vtkoutputpath, rotation, translation, interpolation):
  ...

我正试图从the transformJ plugin sources(见here to understand how it works

中获取灵感

我想使用vtk.vtkTransform,但我真的不明白它是如何工作的:这些examples与我想做的不够接近。这就是我对此所做的:

reader = vtk.vtkXMLImageDataReader()
reader.SetFileName(vtkinputpath)
reader.Update()
transform = reader.vtkTransform()
transform.RotateX(rotation[0])
transform.RotateY(rotation[1])
transform.RotateZ(rotation[2])
transform.Translate(translation[0], translation[1], translation[2])
#and I don't know how I can choose the parameter of the interpolation

但这不起作用...... 我看到here函数RotateWXYZ()存在:

# create a transform that rotates the cone
transform = vtk.vtkTransform()
transform.RotateWXYZ(45,0,1,0)
transformFilter=vtk.vtkTransformPolyDataFilter()
transformFilter.SetTransform(transform)
transformFilter.SetInputConnection(source.GetOutputPort())
transformFilter.Update()

但我不明白这些线是做什么的。 我的主要问题是我找不到Python的vtk文档...

你能告诉我一个Python的vtk文档网站吗?或者你能解释一下vtktransform(rotateWXYZ())的工作原理吗? 拜托,我完全迷失了,没有任何作用。

1 个答案:

答案 0 :(得分:1)

我不确定是否有特定的Python文档,但这对于了解RotateWXYZ的工作原理非常有用:http://www.vtk.org/doc/nightly/html/classvtkTransform.html#a9a6bcc6b824fb0a9ee3a9048aa6b262c

要创建您想要的变换,您可以将旋转矩阵和平移矩阵组合成4x4矩阵,为此我们将旋转矩阵放在列0和行0中,我们将平移向量放在右列中,底行是0,0,0,1。 Here's some more info about this。例如:

0.61782155   0.78631834  0      -0.33657291
0.78631834  -0.61782155  0       1.04497454
0            0          -1       0
0            0           0       1

然后你可以使用SetMatrix直接将矩阵设置为vtkTransform:

matrix = [0.61782155,0.78631834,0,-0.33657291,0.78631834,-0.61782155,0,1.04497454,0,0,-1,0,0,0,0,1]
transform.SetMatrix(matrix)

编辑:编辑完成矩阵变量中的值。