INNER CODE UNIT · Python
shs_view
xverse-engine/XScene-UEPlugin · Training/gaussian-splatting/gaussian_renderer/__init__.py:77
shs_view = pc.get_features.transpose(1, 2).view(-1, 3, (pc.max_sh_degree+1)**2)
dir_pp = (pc.get_xyz - viewpoint_camera.camera_center.repeat(pc.get_features.shape[0], 1))
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:
if separate_sh:
dc, shs = pc.get_features_dc, pc.get_features_rest
else:
shs = pc.get_features
else:
colors_precomp = override_color
# Rasterize visible Gaussians to image, obtain their radii (on screen).
if separate_sh:
rendered_image, radii, depth_image = rasterizer(
means3D = means3D,
means2D = means2D,