INNER CODE UNIT · Python
dir_pp_normalized
muskie82/MonoGS · gaussian_splatting/gaussian_renderer/__init__.py:107
dir_pp_normalized = dir_pp / dir_pp.norm(dim=1, keepdim=True)
sh2rgb = eval_sh(pc.active_sh_degree, shs_view, dir_pp_normalized)
colors_precomp = torch.clamp_min(sh2rgb + 0.5, 0.0)
else:
shs = pc.get_features
else:
colors_precomp = override_color
# Rasterize visible Gaussians to image, obtain their radii (on screen).
if mask is not None:
rendered_image, radii, depth, opacity = rasterizer(
means3D=means3D[mask],
means2D=means2D[mask],
shs=shs[mask],
colors_precomp=colors_precomp[mask] if colors_precomp is not None else None,
opacities=opacity[mask],
scales=scales[mask],
rotations=rotations[mask],