You need to sign in or sign up before continuing.
Commit fdcf3687 authored by Jeremy Reizenstein's avatar Jeremy Reizenstein Committed by Facebook GitHub Bot
Browse files

import vis parts separately

Summary: We envision `pytorch3d.vis` to contain submodules with different dependencies. Allow (and require) them to be imported independently.

Reviewed By: theschnitz

Differential Revision: D24622519

fbshipit-source-id: 44840f70f5fd2bd410405bf09546024e48238744
parent 0e5f4f76
...@@ -67,7 +67,7 @@ ...@@ -67,7 +67,7 @@
"\n", "\n",
"# Data structures and functions for rendering\n", "# Data structures and functions for rendering\n",
"from pytorch3d.structures import Pointclouds\n", "from pytorch3d.structures import Pointclouds\n",
"from pytorch3d.vis import AxisArgs, plot_batch_individually, plot_scene\n", "from pytorch3d.vis.plotly_vis import AxisArgs, plot_batch_individually, plot_scene\n",
"from pytorch3d.renderer import (\n", "from pytorch3d.renderer import (\n",
" look_at_view_transform,\n", " look_at_view_transform,\n",
" FoVOrthographicCameras, \n", " FoVOrthographicCameras, \n",
......
...@@ -93,7 +93,8 @@ ...@@ -93,7 +93,8 @@
"\n", "\n",
"# Data structures and functions for rendering\n", "# Data structures and functions for rendering\n",
"from pytorch3d.structures import Meshes\n", "from pytorch3d.structures import Meshes\n",
"from pytorch3d.vis import AxisArgs, plot_batch_individually, plot_scene, texturesuv_image_matplotlib\n", "from pytorch3d.vis.plotly_vis import AxisArgs, plot_batch_individually, plot_scene\n",
"from pytorch3d.vis.texture_vis import texturesuv_image_matplotlib\n",
"from pytorch3d.renderer import (\n", "from pytorch3d.renderer import (\n",
" look_at_view_transform,\n", " look_at_view_transform,\n",
" FoVPerspectiveCameras, \n", " FoVPerspectiveCameras, \n",
......
# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved. # Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
from .plotly_vis import AxisArgs, Lighting, plot_batch_individually, plot_scene
from .texture_vis import texturesuv_image_matplotlib, texturesuv_image_PIL
__all__ = [k for k in globals().keys() if not k.startswith("_")]
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment