Fix bug in SplatfactoModel when sh_degree is zero the color is not wi…

…th sigmoid (#3207) * torch.compile -> torch_compile from misc * sigmoid on color when sh_degree==0 --------- Co-authored-by: Ruilong Li <[email protected]>
nerfstudio-project · Jun 10, 2024 · 4b61cb9 · 4b61cb9
1 parent d220600
commit 4b61cb9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/nerfstudio/models/splatfacto.py b/nerfstudio/models/splatfacto.py
@@ -728,6 +728,7 @@ def get_outputs(self, camera: Cameras) -> Dict[str, Union[torch.Tensor, List]]:
         if self.config.sh_degree > 0:
             sh_degree_to_use = min(self.step // self.config.sh_degree_interval, self.config.sh_degree)
         else:
+            colors_crop = torch.sigmoid(colors_crop)
             sh_degree_to_use = None
 
         render, alpha, info = rasterization(