Python 三维网格处理库 Trimesh 详解 | 极客日志

PythonAI算法

Python 三维网格处理库 Trimesh 详解

Trimesh 是一个用于处理和操作三维网格的 Python 库，支持 STL、OBJ 等多种格式加载与导出。它提供模型变换、网格分析（体积、面积）、布尔运算、简化优化、碰撞检测及点云体素化等功能。结合 NumPy 使用，适用于 3D 打印、计算机视觉及物理仿真场景，并具备内置可视化工具。

追风少年发布于 2026/3/30更新于 2026/4/131 浏览

Python 三维网格处理库 Trimesh 详解

1. Trimesh 是什么？

Trimesh 是一个用于处理和操作三维网格（mesh）的 Python 库。它提供了一组强大且易于使用的工具，用于加载、编辑、分析和可视化三维网格数据。

Trimesh 支持多种网格格式（如 STL, OBJ, PLY, GLTF 等），并且在计算机图形学和几何计算中非常有用，特别适合用于 3D 模型的操作、分析和可视化。

2. 安装与基础设置

使用以下命令安装 Trimesh 库：

pip install trimesh

pip install pythreejs vtk

import trimesh
import numpy as np

# 读取 3D 模型文件
mesh = trimesh.load('path_to_your_model.obj')

# 打印模型信息
print(f"顶点数量：{len(mesh.vertices)}")
print(f"面数量：{len(mesh.faces)}")

# 保存模型
mesh.export('path_to_save_model.stl')

# 平移模型
mesh.apply_translation([1,0,0])

# 旋转模型
mesh.apply_rotation([0,0, np.pi /4])

# 缩放模型
mesh.apply_scale(2)

# 或者使用变换矩阵
rotation_matrix = np.array([
    [1,0,0,0],
    [0,0.866,-0.5,0],
    [0,0.5,0.866,0],
    [0,0,0,1]
])
mesh.apply_transform(rotation_matrix)

# 计算网格的基本属性
volume = mesh.volume      # 获取网格的体积
surface_area = mesh.area  # 表面积
is_watertight = mesh.is_watertight  # 检查网格是否为闭合网格

print(f"体积：{volume:.2f}")
print(f"表面积：{surface_area:.2f}")
print(f"是否闭合：{is_watertight}")

# 获取边界框
bounding_box = mesh.bounding_box
print(f"边界框：{bounding_box}")

# 检查网格是否是凸的
is_convex = mesh.is_convex

# 加载两个三维模型
mesh1 = trimesh.load('path_to_model1.stl')
mesh2 = trimesh.load('path_to_model2.stl')

# 合并三维模型
combined_mesh = mesh1 + mesh2

# 布尔运算
union_mesh = mesh1.union(mesh2)       # 并集
difference_mesh = mesh1.difference(mesh2)  # 差集
intersection_mesh = mesh1.intersection(mesh2)  # 交集

# 去除重复顶点
combined_mesh.merge_vertices()

# 网格简化
simplified_mesh = mesh.simplify_quadratic_decimation(face_count=1000)

# 网格细分
subdivided_mesh = mesh.subdivide()

# 网格平滑
smoothed_mesh = mesh.smooth_laplacian()

# 网格修复（如果网格有问题）
mesh.fill_holes()

# 创建新的 3D 模型
new_mesh = trimesh.primitives.Sphere(radius=1.0)

# 检查两个模型是否碰撞
collision = mesh.intersects(new_mesh)
print(f"是否碰撞：{collision}")

# 精确的碰撞检测
collision_manager = trimesh.collision.CollisionManager()
collision_manager.add_object('mesh1', mesh)
collision_manager.add_object('mesh2', new_mesh)

# 检测管理器中所有对象的碰撞
is_colliding = collision_manager.in_collision_internal()

# 从网格表面采样点云
points = mesh.sample(2048)

# 体素化网格
voxel_size = 0.01
voxel_grid = mesh.voxelized(voxel_size)

# 或者使用另一种体素化方法
voxel = trimesh.voxel.Voxel(mesh, size=voxel_size)
voxel_data = voxel.data

# 将体素转换为网格
voxel_mesh = voxel_grid.as_boxes()

# 显示体素
voxel_mesh.show()

# 基本可视化
mesh.show()

# 在场景中可视化多个网格
scene = trimesh.Scene()
scene.add_geometry(mesh)
scene.add_geometry(trimesh.primitives.Sphere(center=[2,2,2]))
scene.show()

# 创建场景对象
scene = trimesh.Scene()

# 加载模型
scene.add_geometry(trimesh.load_mesh('model.obj'))

# 设置窗口大小和位置
viewer = trimesh.viewer.windowed(scene, width=800, height=600, x=100, y=100)

# 显示窗口
viewer.show()

# 使用 VTK 进行高级渲染 (示例)
import vtk

# 创建 VTK 渲染管道
reader = vtk.vtkSTLReader()
reader.SetFileName('path_to_your_model.stl')
reader.Update()

mapper = vtk.vtkPolyDataMapper()
mapper.SetInputConnection(reader.GetOutputPort())

actor = vtk.vtkActor()
actor.SetMapper(mapper)

renderer = vtk.vtkRenderer()
render_window = vtk.vtkRenderWindow()
render_window.AddRenderer(renderer)

render_window_interactor = vtk.vtkRenderWindowInteractor()
render_window_interactor.SetRenderWindow(render_window)

renderer.AddActor(actor)
renderer.SetBackground(0.1,0.2,0.4)

render_window.Render()
render_window_interactor.Start()

import trimesh
import numpy as np

def process_3d_model(input_path, output_path):
    # 加载模型
    mesh = trimesh.load(input_path)
    
    # 检查模型信息
    print(f"原始模型 - 顶点数：{len(mesh.vertices)}, 面数：{len(mesh.faces)}")
    
    # 检查并修复网格
    if not mesh.is_watertight:
        print("网格不是封闭的，尝试修复...")
        mesh.fill_holes()
    
    # 简化网格（如果面数太多）
    if len(mesh.faces) > 10000:
        target_faces = 10000
        mesh = mesh.simplify_quadratic_decimation(target_faces)
    
    print(f"简化后 - 顶点数：{len(mesh.vertices)}, 面数：{len(mesh.faces)}")
    
    # 应用变换
    mesh.apply_translation([0,0,0])  # 移动到原点
    
    # 计算边界框并居中
    mesh.apply_translation(-mesh.centroid)
    
    # 保存处理后的模型
    mesh.export(output_path)
    print(f"处理后的模型已保存到：{output_path}")
    return mesh

# 使用示例
processed_mesh = process_3d_model('input.obj','output.obj')
processed_mesh.show()

# 点云分类数据准备示例
def prepare_point_cloud_data(mesh_path, num_points=2048):
    """ 为点云分类准备数据 """
    # 加载网格
    mesh = trimesh.load(mesh_path)
    
    # 从网格表面采样点
    points = mesh.sample(num_points)
    
    # 计算法线（如果需要）
    mesh.face_normals
    
    return points

# 使用示例
points = prepare_point_cloud_data('model.obj')
print(f"点云形状：{points.shape}")

# 可视化点云
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D

fig = plt.figure(figsize=(5,5))
ax = fig.add_subplot(111, projection="3d")
ax.scatter(points[:,0], points[:,1], points[:,2])
ax.set_axis_off()
plt.show()

def optimize_large_mesh(mesh, target_faces=None):
    """ 优化大型网格的性能 """
    original_faces = len(mesh.faces)
    
    # 如果未指定目标面数，使用启发式方法
    if target_faces is None:
        if original_faces > 100000:
            target_faces = original_faces // 10
        elif original_faces > 50000:
            target_faces = original_faces // 5
        else:
            target_faces = original_faces
    
    # 简化网格
    if target_faces < original_faces:
        mesh = mesh.simplify_quadratic_decimation(target_faces)
    
    # 合并顶点
    mesh.merge_vertices()
    
    print(f"网格从 {original_faces} 面简化到 {len(mesh.faces)} 面")
    return mesh

# 使用示例
large_mesh = trimesh.load('large_model.ply')
optimized_mesh = optimize_large_mesh(large_mesh)

# 分批处理大型模型
def process_large_model_in_chunks(model_path, chunk_callback):
    """ 分批处理大型模型 """
    mesh = trimesh.load(model_path)
    
    # 如果面数太多，分割处理
    if len(mesh.faces) > 100000:
        # 使用边界框分割
        split_meshes = mesh.split()
        for i, submesh in enumerate(split_meshes):
            print(f"处理第 {i+1}/{len(split_meshes)} 部分...")
            chunk_callback(submesh, i)
    else:
        chunk_callback(mesh, 0)

Python 三维网格处理库 Trimesh 详解

Python 三维网格处理库 Trimesh 详解

1. Trimesh 是什么？

2. 安装与基础设置

3. 核心功能与应用

3.1 加载与导出 3D 模型

3.2 模型变换与操作

3.3 网格分析和属性计算

3.4 模型合并与布尔运算

3.5 网格简化与优化

3.6 碰撞检测与物理仿真

3.7 点云与体素化

4. 可视化与场景管理

4.1 基本可视化

4.2 自定义可视化窗口

4.3 与其他可视化库集成

5. 实际应用案例

5.1 3D 模型处理流程

5.2 在计算机视觉中的应用

6. 性能优化与最佳实践

6.1 处理大型网格

6.2 内存管理

7. 总结

更多推荐文章

相关免费在线工具

Python 三维网格处理库 Trimesh 详解

Python 三维网格处理库 Trimesh 详解

1. Trimesh 是什么？

2. 安装与基础设置

3. 核心功能与应用

3.1 加载与导出 3D 模型

3.2 模型变换与操作

3.3 网格分析和属性计算

3.4 模型合并与布尔运算

3.5 网格简化与优化

3.6 碰撞检测与物理仿真

3.7 点云与体素化

4. 可视化与场景管理

4.1 基本可视化

4.2 自定义可视化窗口

4.3 与其他可视化库集成

5. 实际应用案例

5.1 3D 模型处理流程

5.2 在计算机视觉中的应用

6. 性能优化与最佳实践

6.1 处理大型网格

6.2 内存管理

7. 总结

微信扫一扫，关注极客日志

更多推荐文章

相关免费在线工具