基于锐度分割3D网格后无法保存分块模型的技术求助

问题根源 #

你的代码中vtkConnectivityFilter使用SetExtractionModeToAllRegions()时，输出的是所有连通区域合并后的整体网格，而非单独的区域。调用GetOutput(i)无法获取第i个独立区域，这是导致无法保存单个网格块的核心原因。

修正后的VTK代码 #

需要在循环中逐个指定要提取的区域，调整提取模式并重新更新过滤器：

import vtk

# 加载输入网格
reader = vtk.vtkSTLReader()
reader.SetFileName("mesh.stl")
reader.Update()

# 计算法向量并分割特征边
mesh = reader.GetOutput()
normals = vtk.vtkPolyDataNormals()
normals.SetInputData(mesh)
normals.ComputePointNormalsOff()
normals.ComputeCellNormalsOn()
angle_threshold = 80  # 角度阈值
normals.SetFeatureAngle(angle_threshold)
normals.SetSplitting(True)
normals.Update()

# 提取连通区域并统计总数
connectivity = vtk.vtkConnectivityFilter()
connectivity.SetInputData(normals.GetOutput())
connectivity.SetExtractionModeToAllRegions()
connectivity.Update()
region_count = connectivity.GetNumberOfExtractedRegions()
print(f"检测到{region_count}个区域")

# 逐个保存每个区域
for i in range(region_count):
    # 切换到指定区域提取模式
    connectivity.SetExtractionModeToSpecifiedRegion()
    connectivity.SetSpecifiedRegion(i)
    connectivity.Update()
    
    # 写入单个区域文件
    writer = vtk.vtkSTLWriter()
    writer.SetFileName(f"output_{i}.stl")
    writer.SetInputData(connectivity.GetOutput())
    writer.Write()

替代方案：使用trimesh库（更简洁） #

如果VTK的流程过于繁琐，可以试试Python的trimesh库，它处理网格分割和保存更直观：

首先安装依赖：

pip install trimesh

基础分割代码（自动按连通性分割）：

import trimesh

# 加载STL网格
mesh = trimesh.load("mesh.stl")

# 分割连通区域
split_meshes = mesh.split(only_watertight=False)

# 保存每个分割后的网格
for idx, sub_mesh in enumerate(split_meshes):
    sub_mesh.export(f"output_trimesh_{idx}.stl")

如果需要基于法线角度先标记特征边再分割：

import trimesh
import numpy as np

mesh = trimesh.load("mesh.stl")
angle_threshold = np.radians(80)  # 转换为弧度

# 计算相邻面的法线夹角
adjacent_faces = mesh.face_adjacency
normals = mesh.face_normals
angles = np.arccos(np.clip(np.dot(normals[adjacent_faces[:,0]], normals[adjacent_faces[:,1]].T), -1, 1))

# 标记角度超阈值的边为边界
boundary_edges = mesh.face_adjacency_edges[angles > angle_threshold]
mesh.edges_unique[boundary_edges, :] = True

# 分割网格
split_meshes = mesh.split(only_watertight=False)

# 保存结果
for idx, sub_mesh in enumerate(split_meshes):
    sub_mesh.export(f"output_trimesh_angle_{idx}.stl")

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