Refactor plot_meshes and plot_pointclouds to one generalizable API, p…

…lot_scene Summary: Defines a function plot_scene that takes in a dictionary defining subplot and trace layouts for Mesh/Pointcloud objects and plots them. Also supports other plotly axis arguments and mesh lighting. Plot_batch_individually is a wrapper function that takes in one or multiple batched Meshes/Pointclouds and uses plot_scene to plot each element within a batch in an individual subplot, possibly sharing that subplot with traces of other individual elements of the other batched structures passed in. Reviewed By: nikhilaravi Differential Revision: D24235479 fbshipit-source-id: 9f669f1b186d55fe5c75552083316c0cf1387472
facebookresearch · Oct 21, 2020 · 964893c · 964893c
1 parent abd3903
commit 964893c
Show file tree

Hide file tree

Showing 2 changed files with 304 additions and 195 deletions.
diff --git a/pytorch3d/vis/__init__.py b/pytorch3d/vis/__init__.py
@@ -1,6 +1,6 @@
 # Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
 
-from .plotly_vis import AxisArgs, Lighting, plot_meshes, plot_pointclouds
+from .plotly_vis import AxisArgs, Lighting, plot_scene
 
 
 __all__ = [k for k in globals().keys() if not k.startswith("_")]