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()
相关推荐
“AI国潮设计-小江”5 小时前
【AIGC实战】Python+SDXL批量生成“潮汕英歌舞不倒翁”IP | 角色一致性与多场景融合方案,附完整代码
开发语言·人工智能·python·prompt·aigc
Linux_cy5 小时前
服务器管理工具哪个好用?我用了七八个
运维·服务器·javascript·python·ssh·面板·gmssh
MoRanzhi12035 小时前
第6篇(下)· 使用 PyQt5 实现 2048 高性能计算版:统一运行控制与跨模式安全交接
python·pyqt·进程管理·状态机·软件架构·并发控制·数据一致性
code tsunami6 小时前
【无标题】
开发语言·python
CodeGL6 小时前
10、Python基础之进程和线程
开发语言·python
TechWayfarer7 小时前
PHP漏洞被利用,怎么揪出攻击IP?
开发语言·python·tcp/ip·网络安全·php
2601_967980118 小时前
Python列表操作方法详解
python·数据类型·修改·列表操作·功能汇总
Patrick在香港8 小时前
台风在变少,来香港的没有
python·数据分析·香港·开放数据·热带气旋
qq_406981038 小时前
大白话rust系列09 函数调用两次有什么错?
开发语言·python·rust
benchmark_cc9 小时前
QuantDash 安装成功后,如何验证 Python 环境可以正常导入 SDK?
开发语言·python·数据分析·pandas·量化交易·股票数据·quantdash