magistrkoljan commited on
Commit
f1adf63
·
verified ·
1 Parent(s): e543d95

Update submodules/gaussian-splatting/gaussian_renderer/__init__.py

Browse files
submodules/gaussian-splatting/gaussian_renderer/__init__.py CHANGED
@@ -88,7 +88,7 @@ def render(viewpoint_camera, pc : GaussianModel, pipe, bg_color : torch.Tensor,
88
 
89
  # Rasterize visible Gaussians to image, obtain their radii (on screen).
90
  if separate_sh:
91
- rendered_image, _, _ = rasterizer(
92
  means3D = means3D,
93
  means2D = means2D,
94
  dc = dc,
@@ -99,7 +99,7 @@ def render(viewpoint_camera, pc : GaussianModel, pipe, bg_color : torch.Tensor,
99
  rotations = rotations,
100
  cov3D_precomp = cov3D_precomp)
101
  else:
102
- rendered_image, _, _ = rasterizer(
103
  means3D = means3D,
104
  means2D = means2D,
105
  shs = shs,
 
88
 
89
  # Rasterize visible Gaussians to image, obtain their radii (on screen).
90
  if separate_sh:
91
+ rendered_image, radii, _ = rasterizer(
92
  means3D = means3D,
93
  means2D = means2D,
94
  dc = dc,
 
99
  rotations = rotations,
100
  cov3D_precomp = cov3D_precomp)
101
  else:
102
+ rendered_image, radii, _ = rasterizer(
103
  means3D = means3D,
104
  means2D = means2D,
105
  shs = shs,