updated comment

nerfstudio-project · Jan 27, 2025 · 2d95d1e · 2d95d1e
1 parent abff91c
commit 2d95d1e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nerfstudio/models/splatfacto.py b/nerfstudio/models/splatfacto.py
@@ -569,8 +569,8 @@ def get_outputs(self, cameras: Cameras) -> Dict[str, Union[torch.Tensor, List]]:
             scales=torch.exp(scales_crop),
             opacities=torch.sigmoid(opacities_crop).squeeze(-1),
             colors=colors_crop,
-            viewmats=viewmats,  # [1, 4, 4]
-            Ks=Ks,  # [1, 3, 3]
+            viewmats=viewmats,  # [B, 4, 4]
+            Ks=Ks,  # [B, 3, 3]
             width=W,
             height=H,
             packed=False,