如何使用Python实现.mha转.stl及3D体数据到表面网格的转换

两种常用实现方式如下：

方案1：使用vedo库（最简实现） #

vedo库已经封装了体数据处理的全流程，仅需几行代码即可完成转换

1. 安装依赖：
   pip install vedo
2. 示例代码：

from vedo import load, write

# 读取mha格式体数据
vol = load("输入文件路径.mha")
# 提取等值面生成网格，iso参数为目标提取区域的灰度阈值，需根据实际数据调整
mesh = vol.isosurface(iso=100)
# 可选：对网格做平滑处理，减少表面锯齿
mesh = mesh.smooth(n_iter=30)
# 导出为stl格式
write(mesh, "输出文件路径.stl")

方案2：使用原生VTK库（灵活定制） #

适合需要自定义处理全流程的场景，直接调用VTK底层接口实现

1. 安装依赖：
   pip install vtk
2. 示例代码：

import vtk

# 读取mha体数据
reader = vtk.vtkMetaImageReader()
reader.SetFileName("输入文件路径.mha")
reader.Update()

# 移动立方体算法提取等值面
marching_cubes = vtk.vtkMarchingCubes()
marching_cubes.SetInputConnection(reader.GetOutputPort())
# 设置等值面阈值，根据体数据灰度范围调整
marching_cubes.SetValue(0, 100)
marching_cubes.Update()

# 可选：网格平滑处理
smoother = vtk.vtkSmoothPolyDataFilter()
smoother.SetInputConnection(marching_cubes.GetOutputPort())
smoother.SetNumberOfIterations(50)
smoother.SetRelaxationFactor(0.1)
smoother.Update()

# 导出stl文件
stl_writer = vtk.vtkSTLWriter()
stl_writer.SetFileName("输出文件路径.stl")
stl_writer.SetInputConnection(smoother.GetOutputPort())
stl_writer.Write()

注意事项 #

• 等值面提取的阈值需要根据你的体数据灰度范围自定义调整，如果不确定合适阈值，可先可视化体数据查看目标区域的灰度区间
• 若生成的网格存在大量冗余噪点面片，可先对体数据做高斯滤波预处理，或增加网格去重、裁剪步骤
• 高分辨率体数据可先做下采样处理，降低计算量和最终stl文件的体积

内容的提问来源于stack exchange，提问作者blackcat