fix viewdir checking in ngp.py

nerfstudio-project · Apr 24, 2024 · 5e08b62 · 5e08b62
1 parent 5cea4fe
commit 5e08b62
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/radiance_fields/ngp.py b/examples/radiance_fields/ngp.py
@@ -212,8 +212,8 @@ def forward(
             assert (
                 positions.shape == directions.shape
             ), f"{positions.shape} v.s. {directions.shape}"
-            density, embedding = self.query_density(positions, return_feat=True)
-            rgb = self._query_rgb(directions, embedding=embedding)
+        density, embedding = self.query_density(positions, return_feat=True)
+        rgb = self._query_rgb(directions, embedding=embedding)
         return rgb, density  # type: ignore