PyTorch3D

PyTorch3D

  • 文档
  • 教程
  • API
  • GitHub

›可视化

简介

  • 为什么选择 PyTorch3D

数据

  • 文件 I/O
  • 从文件加载
  • 数据加载器
  • 批处理

操作

  • Cubify
  • IoU3D

可视化

  • Plotly 可视化

渲染器

  • 概述
  • 入门
  • 相机

概述

PyTorch3D 提供了一个模块化的可微渲染器,但对于我们想要交互式绘图或不关心渲染过程的可微性的情况,我们提供了在 Plotly 中渲染网格和点云的函数。这些 Plotly 图形允许您旋转和缩放渲染的图像,并支持将批处理数据作为单个绘图中的多个轨迹或划分为单独的子图进行绘制。

示例

这些渲染函数接受 Plotly 的 x、y 和 z 轴参数作为kwargs,允许我们自定义绘图。这里有两个带彩色轴的绘图,一个点云绘图,一个子图中的批处理网格绘图,以及一个具有多个轨迹的批处理网格绘图。请参阅渲染纹理网格和渲染彩色点云教程以获取代码示例。

将绘图保存为图像

如果您想保存这些 Plotly 绘图,则需要安装一个单独的库,例如Kaleido。

安装 Kaleido

$ pip install Kaleido

将图形导出为 .png 图像。图像将保存到当前工作目录中。

fig = ...
fig.write_image("image_name.png")
← IoU3D概述 →
pytorch3d
Facebook Open Source
版权所有 © 2024 Meta Platforms, Inc
法律:隐私条款