Hi3DGen

Running on Zero

Stable-X commited on 15 days ago

Commit

550f807

1 Parent(s): 1ab5d3a

fix: Add color rendering

Files changed (1) hide show

trellis/renderers/mesh_renderer.py CHANGED Viewed

@@ -64,7 +64,7 @@ class MeshRenderer:
             mesh : MeshExtractResult,
             extrinsics: torch.Tensor,
             intrinsics: torch.Tensor,
-            return_types = ["mask", "normal", "depth"]
         ) -> edict:
         """
         Render the mesh.

             mesh : MeshExtractResult,
             extrinsics: torch.Tensor,
             intrinsics: torch.Tensor,
+            return_types = ["mask", "normal", "depth", "color"]
         ) -> edict:
         """
         Render the mesh.