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()