12

在 VTK 中,我可以使用以下代码段将渲染窗口保存为图像。但是,实际上我想直接将它作为一个 numpy 数组获取(无需写入然后读取)。

im = vtkWindowToImageFilter()
writer = vtkPNGWriter()
im.SetInput(renderWindow)
im.Update()
writer.SetInputConnection(im.GetOutputPort())
writer.SetFileName("file.png")
writer.Write()

做这个的最好方式是什么?

4

1 回答 1

12

我相信没有必要涉及vtkXWriterX某种格式在哪里),除非您需要X格式中的数据。在定义了要将其内容导出为图像的窗口后,您可以继续获取 VTK 图像并使用它。

from vtk.util.numpy_support import vtk_to_numpy

...

vtk_rw = vtk.vtkRenderWindow()

...

vtk_win_im = vtk.vtkWindowToImageFilter()
vtk_win_im.SetInput(vtk_rw)
vtk_win_im.Update()

vtk_image = vtk_win_im.GetOutput()

width, height, _ = vtk_image.GetDimensions()
vtk_array = vtk_image.GetPointData().GetScalars()
components = vtk_array.GetNumberOfComponents()

arr = vtk_to_numpy(vtk_array).reshape(height, width, components)

...
于 2013-01-28T18:40:44.670 回答