Python从VTK 的 RenderWindow中取出 JPG图像信息变成 numpy格式

comedate

已于 2022-03-15 16:02:20 修改

阅读量704

点赞数

分类专栏： Python实用源码 VTK 文档枝术文章标签： python VTK JPG图像 numpy WindowToImage

于 2020-10-30 22:42:37 首次发布

本文链接：https://blog.csdn.net/comedate/article/details/109352210

版权

Python实用源码同时被 3 个专栏收录

106 篇文章 2 订阅

订阅专栏

文档枝术

27 篇文章 0 订阅

订阅专栏

VTK

25 篇文章 2 订阅

订阅专栏

该博客介绍了如何使用VTK库将渲染窗口的图像转换为JPEG格式，以便在QT窗口中显示或通过Base64编码发送到Chrome浏览器。主要涉及的函数包括vtkWindowToImageFilter、vtkJPEGWriter以及vtk_to_numpy，实现了从VTK渲染窗口到numpy数组再到字节流的转换过程。

摘要由CSDN通过智能技术生成

前言

在使用VTK显示图像时，默认方式是将图像显示到 Window 上。但是，有一种情况，我们需要将图像变成JPG，用于显示在QT的窗口或者 Base64压缩后发送到 Chrome 的浏览器的前端去。
因此，写如下的函数接口。

主要使用函数：

vtkWindowToImageFilter
vtkJPEGWriter
vtk_to_numpy

代码如下：

#!/usr/bin/python
# -*- coding: UTF-8 -*-
import time
import vtk
from vtk.util.numpy_support import vtk_to_numpy


def render_result_to_bytes(ren_win, quality):
    """ 将render_window的结果输出成 numpy 格式 """
    begin_time = time.time()

    window_to_image_filter = vtk.vtkWindowToImageFilter()
    window_to_image_filter.SetInput(ren_win)
    window_to_image_filter.SetInputBufferTypeToRGB()
    window_to_image_filter.Update()

    writer = vtk.vtkJPEGWriter()
    writer.SetInputConnection(window_to_image_filter.GetOutputPort())
    writer.SetQuality(quality)
    writer.WriteToMemoryOn()
    writer.Write()
	
	vtk_data = writer.GetResult()
    data = vtk_to_numpy(vtk_data).tobytes()

    print(r'render_result_to_bytes cost time{}'.format(time.time() - begin_time))
    return data