### 使用 `trimesh` 和 `pyrender` 绘制坐标系 `trimesh` 提供了创建 3D 坐标系的功能，可以通过 `trimesh.creation.axis` 生成一个包含 X、Y、Z 三个轴的线段模型，同时支持设置长度和原点标记大小 [^1]。为了实现更高质量的 3D 可视化，可以结合 `pyrender` 进行渲染。`pyrender` 是一个基于 OpenGL 的轻量级 3D 渲染库，适用于在 Python 中进行 3D 场景的构建和可视化 [^1]。 #### 创建坐标系模型 使用 `trimesh.creation.axis` 创建一个标准的坐标系模型，该模型包含三个轴的线段和原点标记。可以通过 `length` 参数调整坐标轴长度，`origin_size` 控制原点标记的大小 [^1]。 ```python import trimesh # 创建坐标系模型 axis = trimesh.creation.axis(length=1.0, origin_size=0.05) ``` #### 将 `trimesh` 模型转换为 `pyrender` 场景 `pyrender` 通过 `Scene` 对象管理 3D 场景，并支持添加 `Mesh` 对象进行渲染。可以将 `trimesh` 生成的坐标系模型转换为 `pyrender.Mesh` 对象，并添加到场景中 [^1]。 ```python import pyrender # 将 trimesh 模型转换为 pyrender.Mesh mesh = pyrender.Mesh.from_trimesh(axis) # 创建场景并添加坐标系 scene = pyrender.Scene() scene.add(mesh) ``` #### 设置相机和光源 为了正确渲染 3D 场景，需要添加相机和光源。`pyrender` 提供了多种相机类型，例如 `PerspectiveCamera`，可以设置视场角（FOV）来控制视角 [^1]。 ```python # 添加相机 camera = pyrender.PerspectiveCamera(yfov=np.pi / 4.0) camera_pose = np.eye(4) camera_pose[:3, 3] = [0, 0, 5] # 设置相机位置 scene.add(camera, pose=camera_pose) # 添加光源 light = pyrender.DirectionalLight(color=[1.0, 1.0, 1.0], intensity=2.0) scene.add(light, pose=camera_pose) ``` #### 渲染并显示图像 使用 `pyrender.Viewer` 或 `pyrender.OffscreenRenderer` 进行渲染。`Viewer` 适用于交互式查看，而 `OffscreenRenderer` 适用于无头环境下的图像渲染 [^1]。 ```python # 使用 Viewer 交互式查看 pyrender.Viewer(scene, use_raymond_lighting=True) ``` 或者使用离屏渲染并显示图像： ```python # 使用 OffscreenRenderer 渲染图像 r = pyrender.OffscreenRenderer(800, 600) color, depth = r.render(scene) import matplotlib.pyplot as plt plt.imshow(color) plt.axis('off') plt.show() ``` #### 完整示例代码 ```python import trimesh import pyrender import numpy as np import matplotlib.pyplot as plt # 创建坐标系模型 axis = trimesh.creation.axis(length=1.0, origin_size=0.05) # 将 trimesh 模型转换为 pyrender.Mesh mesh = pyrender.Mesh.from_trimesh(axis) # 创建场景并添加坐标系 scene = pyrender.Scene() scene.add(mesh) # 添加相机 camera = pyrender.PerspectiveCamera(yfov=np.pi / 4.0) camera_pose = np.eye(4) camera_pose[:3, 3] = [0, 0, 5] # 设置相机位置 scene.add(camera, pose=camera_pose) # 添加光源 light = pyrender.DirectionalLight(color=[1.0, 1.0, 1.0], intensity=2.0) scene.add(light, pose=camera_pose) # 使用 Viewer 交互式查看 pyrender.Viewer(scene, use_raymond_lighting=True) # 或者使用 OffscreenRenderer 渲染图像 r = pyrender.OffscreenRenderer(800, 600) color, depth = r.render(scene) plt.imshow(color) plt.axis('off') plt.show() ``` ---