Refactor plot_meshes and plot_pointclouds to one generalizable API, plot_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

Author: Amitav Baruah

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("_")]