UniqAI/gaussian-splatting
1
0
Fork 0
mirror of https://github.com/graphdeco-inria/gaussian-splatting synced 2024-11-22 08:18:17 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update __init__.py

Browse Source
This commit is contained in:
JonathonLuiten 2023-07-12 20:35:43 -04:00 committed by GitHub
parent d843dc1224
commit 473560e399
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
gaussian_renderer/__init__.py
View File

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