GenerateCubesFromLabels 提取和可视化特定标签所代表的 3D 结构

一:主要的知识点

1、说明

本文只是教程内容的一小段,因博客字数限制,故进行拆分。主教程链接:vtk教程------逐行解析官网所有Python示例-CSDN博客

2、知识点纪要

本段代码主要涉及的有①体素数据读取vtkMetaImageReader的使用,②vtkGeometryFilter三维网格数据转换为可渲染的polydata

二:代码及注释

python 复制代码
import vtkmodules.vtkRenderingOpenGL2
import vtkmodules.vtkInteractionStyle
from vtkmodules.vtkCommonColor import vtkNamedColors
from vtkmodules.vtkIOImage import vtkMetaImageReader
from vtkmodules.vtkImagingCore import vtkImageWrapPad
from vtkmodules.vtkFiltersCore import vtkThreshold
from vtkmodules.vtkCommonTransforms import vtkTransform
from vtkmodules.vtkFiltersGeneral import vtkTransformFilter
from vtkmodules.vtkFiltersGeometry import vtkGeometryFilter
from vtkmodules.vtkCommonDataModel import (
    vtkDataObject,
    vtkDataSetAttributes
)
from vtkmodules.vtkRenderingCore import (
    vtkActor,
    vtkPolyDataMapper,
    vtkRenderWindow,
    vtkRenderWindowInteractor,
    vtkRenderer
)

def main():
    colors = vtkNamedColors()

    file_name = "Data/Frog/frogtissue.mhd"
    start_label = 1
    end_label = 29

    reader = vtkMetaImageReader()
    reader.SetFileName(file_name)
    reader.Update()

    """
    vtkMetaImageReader 返回的是一个体积数据(vtkImageData 或 vtkStructuredPoints 的基类)
    返回一个包含 6 个整数的元组或列表 [Imin, Imax, Jmin, Jmax, Kmin, Kmax]
    描述的是 3D 网格在三个维度上的体素/点索引范围
    """
    extent = reader.GetOutput().GetExtent()
    print("extent: ", extent)

    """
    vtkImageWrapPad 主要作用是扩展(或填充)图像的边界,并使用"环绕"(Wrap)模式来决定填充的值
    "Wrap"是它区别于其他填充模式(如常数填充 ConstantPad 或重复填充 ClampPad)的关键
    填充值来源: 当它填充新添加的边界体素时,它会从图像另一侧的现有数据中获取数值
    
    包裹填充的原因:在一些算法中,比如图像卷积(滤波),计算梯度或导数周期性模拟(如流体边界循环)
    需要图像在边界是"周期连续"的,这时候不能用零填充(那会引入边缘效应),而是要从另一侧"包裹"过来
    """
    pad = vtkImageWrapPad()
    pad.SetInputConnection(reader.GetOutputPort())
    pad.SetOutputWholeExtent(extent[0], extent[1] + 1, extent[2], extent[3] + 1, extent[4], extent[5] + 1)
    pad.Update()

    """
    数据属性(标量数据)从点到单元(cell)的"拷贝映射" 操作
    为什么这么做:
    一些滤波器、mapper、渲染器期望标量值在 cell 上;
    而输入的原始数据标量通常存在于 points 上;
    所以必须手动把标量从 PointData 复制到 CellData
    """
    pad.GetOutput().GetCellData().SetScalars(reader.GetOutput().GetPointData().GetScalars())

    """
    vtkThreshold  主要用于对数据集(如 vtkUnstructuredGrid, vtkPolyData, vtkImageData 等)中的标量值进行阈值筛选(Thresholding)
    从而提取出满足某个标量范围条件的单元(cells)或点(points)
    """
    selector = vtkThreshold()
    """
    SetInputArrayToProcess 参数详解
    第0个0: 代表输入端口索引
    第1个0: 代表连接的输入数据索引
    第2个0:表示"时间步索引",一般不用改
    vtkDataObject().FIELD_ASSOCIATION_CELLS  表示要使用单元(cell)数据,而不是点(Point)数据
    vtkDataSetAttributes().SCALARS 表示使用当前单元数据中的 标量数组(Scalars)
    """
    selector.SetInputArrayToProcess(0, 0, 0, vtkDataObject().FIELD_ASSOCIATION_CELLS,
                                    vtkDataSetAttributes().SCALARS)
    selector.SetInputConnection(pad.GetOutputPort())
    selector.SetLowerThreshold(start_label)
    selector.SetUpperThreshold(end_label)
    # 只保留标量值在 [start_label, end_label] 之间的单元
    selector.Update()

    transform = vtkTransform()
    transform.Translate(-0.5, -0.5, -0.5)

    transform_model = vtkTransformFilter()
    transform_model.SetTransform(transform)
    transform_model.SetInputConnection(selector.GetOutputPort())

    """
    vtkGeometryFilter  把 任意数据类型(体数据、结构化网格、非结构化网格等)提取成表面几何数据(vtkPolyData)
    简单来说就是把三维网格或体数据转换为可渲染的表面(PolyData)
    """
    geometry = vtkGeometryFilter()
    geometry.SetInputConnection(transform_model.GetOutputPort())

    mapper = vtkPolyDataMapper()
    mapper.SetInputConnection(geometry.GetOutputPort())
    mapper.SetScalarRange(start_label, end_label)
    mapper.SetScalarModeToUseCellData()
    mapper.SetColorModeToMapScalars()

    actor = vtkActor()
    actor.SetMapper(mapper)

    renderer = vtkRenderer()
    render_window = vtkRenderWindow()
    render_window.AddRenderer(renderer)
    render_window.SetSize(640, 480)
    render_window.SetWindowName('GenerateCubesFromLabels')

    render_window_interactor = vtkRenderWindowInteractor()
    render_window_interactor.SetRenderWindow(render_window)

    renderer.AddActor(actor)
    renderer.SetBackground(colors.GetColor3d('DarkSlateBlue'))
    render_window.Render()

    camera = renderer.GetActiveCamera()
    camera.SetPosition(42.301174, 939.893457, -124.005030)
    camera.SetFocalPoint(224.697134, 221.301653, 146.823706)
    camera.SetViewUp(0.262286, -0.281321, -0.923073)
    camera.SetDistance(789.297581)
    camera.SetClippingRange(168.744328, 1509.660206)

    render_window_interactor.Start()

if __name__ == '__main__':
    main()
相关推荐
wj3055853782 小时前
课程 9:模型测试记录与 Prompt 策略
linux·人工智能·python·comfyui
星寂樱易李3 小时前
iperf3 + Python-- 网络带宽、网速、网络稳定性
开发语言·网络·python
qingfeng154153 小时前
企业微信机器人开发:如何实现自动化与智能运营?
人工智能·python·机器人·自动化·企业微信
彦为君6 小时前
Agent 安全:从权限提示到沙箱隔离
python·ai·ai编程
PILIPALAPENG7 小时前
Python 语法速成指南:前端开发者视角(JS 类比版)
前端·人工智能·python
用户8356290780518 小时前
Python 操作 PowerPoint 页眉与页脚指南
后端·python
枫叶林FYL8 小时前
项目九:异步高性能爬虫与数据采集中枢 —— 基于 Crawl<sub>4</sub>AI 与 Playwright 的现代化数据采集平台 项目总览
爬虫·python·深度学习·wpf
猫猫的小茶馆9 小时前
【Python】函数与模块化编程
linux·开发语言·arm开发·驱动开发·python·stm32
Miss_min9 小时前
128K长序列数据生成
开发语言·python·深度学习