This commit is contained in:
JonathonLuiten
2024-09-02 23:29:31 -04:00
committed by GitHub

View File

@@ -82,7 +82,7 @@ def render(viewpoint_camera, pc : GaussianModel, pipe, bg_color : torch.Tensor,
colors_precomp = override_color
# Rasterize visible Gaussians to image, obtain their radii (on screen).
rendered_image, radii = rasterizer(
rendered_image, radii, depth = rasterizer(
means3D = means3D,
means2D = means2D,
shs = shs,
@@ -96,5 +96,6 @@ def render(viewpoint_camera, pc : GaussianModel, pipe, bg_color : torch.Tensor,
# They will be excluded from value updates used in the splitting criteria.
return {"render": rendered_image,
"viewspace_points": screenspace_points,
"visibility_filter" : radii > 0,
"radii": radii}
"visibility_filter": radii > 0,
"radii": radii,
"depth": depth}