fix viewdir checking in ngp.py (#295)

1e46cb1d · Junchen Liu · GitHub · 5cea4feb · 1e46cb1d
Unverified Commit 1e46cb1d authored Apr 23, 2024 by Junchen Liu Committed by GitHub Apr 23, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

examples/radiance_fields/ngp.py examples/radiance_fields/ngp.py +2 -2

No files found.
--- a/examples/radiance_fields/ngp.py
+++ b/examples/radiance_fields/ngp.py
@@ -212,8 +212,8 @@ class NGPRadianceField(torch.nn.Module):
            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