Python VTK将窗口图像渲染到numpy阵列

在VTK中，我可以使用以下代码片段将渲染窗口保存为图像。但是，实际上我想直接将其作为numpy数组（不需要先写后读）获取

---

最好的方法是什么？

我认为没有必要涉及

vtkXWriter

（其中

X

是某种格式），除非您需要

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)

...

---

检查

vtk.util.numpy\u支持

@mmgp-谢谢，但我还没有找到如何从渲染窗口获得可用的内容，我可以将这些内容传递到

vtk\u\u numpy

：(

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)

...