python vtk 图片保存和相机设置。

本文档记录了使用PythonVTK库调整相机视角并保存渲染图片的过程。首先创建vtkRenderer和设置背景颜色,然后通过GetActiveCamera()获取相机并设置其位置、视角等参数。接着,利用vtkWindowToImageFilter和各种图像文件写入器将渲染窗口保存为不同格式的图片。详细步骤和代码示例可供参考。
摘要由CSDN通过智能技术生成

Python VTK中文资料好少啊,有个任务要调整摄像机位置并且保存图片,中文帖子没几个这方面的,啃官方文档解决啦,粗略记录一下。
调整相机:

render = vtk.vtkRenderer()
render.AddActor(actor)
render.SetBackground(0, 0, 0)
camera = render.GetActiveCamera()
#设置视角
camera.SetPosition(0, 60, 500)
camera.SetFocalPoint(0, 100, 0)
#设置物体
camera.Azimuth(60)
camera.Elevation(-300)
camera.Dolly(1.5)
camera.Roll(11)
camera.Pitch(11)
camera.Zoom(10)

保存图片:

from pathlib import Path

from vtkmodules.vtkIOImage import (
    vtkBMPWriter,
    vtkJPEGWriter,
    vtkPNGWriter,
    vtkPNMWriter,
    vtkPostScriptWriter,
    vtkTIFFWriter
)
from vtkmodules.vtkRenderingCore import vtkWindowToImageFilter


def write_image(file_name, ren_win, rgba=True):
    """
    Write the render window view to an image file.

    Image types supported are:
     BMP, JPEG, PNM, PNG, PostScript, TIFF.
    The default parameters are used for all writers, change as needed.

    :param file_name: The file name, if no extension then PNG is assumed.
    :param ren_win: The render window.
    :param rgba: Used to set the buffer type.
    :return:
    """

    if file_name:
        valid_suffixes = ['.bmp', '.jpg', '.png', '.pnm', '.ps', '.tiff']
        # Select the writer to use.
        parent = Path(file_name).resolve().parent
        path = Path(parent) / file_name
        if path.suffix:
            ext = path.suffix.lower()
        else:
            ext = '.png'
            path = Path(str(path)).with_suffix(ext)
        if path.suffix not in valid_suffixes:
            print(f'No writer for this file suffix: {ext}')
            return
        if ext == '.bmp':
            writer = vtkBMPWriter()
        elif ext == '.jpg':
            writer = vtkJPEGWriter()
        elif ext == '.pnm':
            writer = vtkPNMWriter()
        elif ext == '.ps':
            if rgba:
                rgba = False
            writer = vtkPostScriptWriter()
        elif ext == '.tiff':
            writer = vtkTIFFWriter()
        else:
            writer = vtkPNGWriter()

        windowto_image_filter = vtkWindowToImageFilter()
        windowto_image_filter.SetInput(ren_win)
        windowto_image_filter.SetScale(1)  # image quality
        if rgba:
            windowto_image_filter.SetInputBufferTypeToRGBA()
        else:
            windowto_image_filter.SetInputBufferTypeToRGB()
            # Read from the front buffer.
            windowto_image_filter.ReadFrontBufferOff()
            windowto_image_filter.Update()

        writer.SetFileName(path)
        writer.SetInputConnection(windowto_image_filter.GetOutputPort())
        writer.Write()
    else:
        raise RuntimeError('Need a filename.')

详细看官方文档
官方文档链接:https://kitware.github.io/vtk-examples/site/

  • 2
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
Python 中使用 VTK(Visualization Toolkit)显示图片,可以按照以下步骤进行操作: 1. 安装 VTK:打开终端或命令提示符,运行以下命令安装 VTK 库: ``` pip install vtk ``` 2. 导入所需模块:在 Python 脚本中导入必要的模块。 ```python import vtk ``` 3. 创建渲染器和交互窗口:创建一个渲染器和交互窗口,用于显示图像。 ```python renderer = vtk.vtkRenderer() renderWindow = vtk.vtkRenderWindow() renderWindow.AddRenderer(renderer) renderWindowInteractor = vtk.vtkRenderWindowInteractor() renderWindowInteractor.SetRenderWindow(renderWindow) ``` 4. 加载图像数据:使用 `vtkPNGReader` 或 `vtkJPEGReader` 等类来加载图像数据。 ```python reader = vtk.vtkPNGReader() reader.SetFileName("path/to/image.png") reader.Update() ``` 5. 创建图像数据源和演员:将加载的图像数据转换为 VTK 图像数据源,并创建对应的演员。 ```python imageSource = vtk.vtkImageActor() imageSource.SetInputData(reader.GetOutput()) ``` 6. 将演员添加到渲染器中:将图像数据的演员添加到渲染器中。 ```python renderer.AddActor(imageSource) ``` 7. 设置渲染器和交互窗口的属性:设置渲染器和交互窗口的一些属性,如背景颜色、交互样式等。 ```python renderer.SetBackground(1.0, 1.0, 1.0) # 设置背景为白色 renderWindow.SetSize(800, 600) # 设置窗口大小 style = vtk.vtkInteractorStyleImage() renderWindowInteractor.SetInteractorStyle(style) # 设置交互样式为图像样式 ``` 8. 开启交互窗口:启动交互窗口,显示图像。 ```python renderWindow.Render() renderWindowInteractor.Start() ``` 以上是使用 VTKPython 中显示图像的基本步骤,你可以根据需要进行进一步的定制和调整。
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值