Minor fix for _RasterizeToPixels back to avoid NaNs (#235)

nerfstudio-project · Jun 22, 2024 · 3b38568 · 3b38568
1 parent 4723fe8
commit 3b38568
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gsplat/cuda/_wrapper.py b/gsplat/cuda/_wrapper.py
@@ -433,15 +433,15 @@ def rasterize_to_pixels(
         colors = torch.cat(
             [
                 colors,
-                torch.empty(*colors.shape[:-1], padded_channels, device=device),
+                torch.zeros(*colors.shape[:-1], padded_channels, device=device),
             ],
             dim=-1,
         )
         if backgrounds is not None:
             backgrounds = torch.cat(
                 [
                     backgrounds,
-                    torch.empty(
+                    torch.zeros(
                         *backgrounds.shape[:-1], padded_channels, device=device
                     ),
                 ],