Python PyTorch SummaryWriter.add_mesh用法及代码示例

本文简要介绍python语言中 torch.utils.tensorboard.writer.SummaryWriter.add_mesh 的用法。

用法:

add_mesh(tag, vertices, colors=None, faces=None, config_dict=None, global_step=None, walltime=None)

参数：

• tag(string) -数据标识符

• vertices(torch.Tensor) -顶点的 3D 坐标列表。

• colors(torch.Tensor) -每个顶点的颜色

• faces(torch.Tensor) -每个三角形内的顶点索引。 (可选的)

• config_dict-带有ThreeJS 类名称和配置的字典。

• global_step(int) -要记录的全局步长 值

• walltime(float) -事件纪元后的可选覆盖默认 walltime (time.time()) 秒

将网格或 3D 点云添加到 TensorBoard。可视化基于 Three.js，因此它允许用户与渲染的对象进行交互。除了顶点、面等基本定义外，用户还可以进一步提供相机参数、光照条件等。请查看https://threejs.org/docs/index.html#manual/en/introduction/Creating-a-scene用于高级用法。

形状：

顶点：(B, N, 3)。 (批次，number_of_vertices，通道)

颜色：(B, N, 3)。对于类型 uint8 的值应位于 [0, 255] 或对于类型 float 的值应位于 [0, 1] 中。

面孔：(B, N, 3)。对于类型 uint8 ，值应位于 [0, number_of_vertices] 中。

例子：

from torch.utils.tensorboard import SummaryWriter
vertices_tensor = torch.as_tensor([
    [1, 1, 1],
    [-1, -1, 1],
    [1, -1, -1],
    [-1, 1, -1],
], dtype=torch.float).unsqueeze(0)
colors_tensor = torch.as_tensor([
    [255, 0, 0],
    [0, 255, 0],
    [0, 0, 255],
    [255, 0, 255],
], dtype=torch.int).unsqueeze(0)
faces_tensor = torch.as_tensor([
    [0, 2, 3],
    [0, 3, 1],
    [0, 1, 2],
    [1, 3, 2],
], dtype=torch.int).unsqueeze(0)

writer = SummaryWriter()
writer.add_mesh('my_mesh', vertices=vertices_tensor, colors=colors_tensor, faces=faces_tensor)

writer.close()