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()
相关推荐
Csvn17 小时前
🌟 LangChain 30 天保姆级教程 · Day 13|OutputParser 进阶!让 AI 输出自动转为结构化对象,并支持自动重试!
python·langchain
cch891818 小时前
Python主流框架全解析
开发语言·python
sg_knight18 小时前
设计模式实战:状态模式(State)
python·ui·设计模式·状态模式·state
好运的阿财18 小时前
process 工具与子agent管理机制详解
网络·人工智能·python·程序人生·ai编程
张張40818 小时前
(域格)环境搭建和编译
c语言·开发语言·python·ai
weixin_4235339918 小时前
【Windows11离线安装anaconda、python、vscode】
开发语言·vscode·python
Ricky111zzz19 小时前
leetcode学python记录1
python·算法·leetcode·职场和发展
小白学大数据19 小时前
Selenium+Python 爬虫:动态加载头条问答爬取
爬虫·python·selenium
Hui Baby19 小时前
springboot读取配置文件
后端·python·flask
阿Y加油吧19 小时前
回溯法经典难题:N 皇后问题 深度解析 + 二分查找入门(搜索插入位置)
开发语言·python