diff --git a/pytorch3d/renderer/__init__.py b/pytorch3d/renderer/__init__.py index 7fca1194..2edce601 100644 --- a/pytorch3d/renderer/__init__.py +++ b/pytorch3d/renderer/__init__.py @@ -49,6 +49,7 @@ from .mesh import ( HardPhongShader, MeshRasterizer, MeshRenderer, + MeshRendererWithFragments, RasterizationSettings, SoftGouraudShader, SoftPhongShader, diff --git a/pytorch3d/renderer/mesh/__init__.py b/pytorch3d/renderer/mesh/__init__.py index 66e422de..ff28349e 100644 --- a/pytorch3d/renderer/mesh/__init__.py +++ b/pytorch3d/renderer/mesh/__init__.py @@ -13,7 +13,7 @@ from .clip import ( ) from .rasterize_meshes import rasterize_meshes from .rasterizer import MeshRasterizer, RasterizationSettings -from .renderer import MeshRenderer +from .renderer import MeshRenderer, MeshRendererWithFragments from .shader import TexturedSoftPhongShader # DEPRECATED from .shader import ( BlendParams,