Pass boolean tensor instead of the occlusion once. (#123)

fe75cb86 · Thomas Wang · GitHub · 0e1e7cb7 · fe75cb86
Unverified Commit fe75cb86 authored Nov 26, 2022 by Thomas Wang Committed by GitHub Nov 27, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

nerfacc/grid.py nerfacc/grid.py +2 -2

No files found.
--- a/nerfacc/grid.py
+++ b/nerfacc/grid.py
@@ -236,7 +236,7 @@ class OccupancyGrid(Grid):
        # )
        self._binary = (
            self.occs > torch.clamp(self.occs.mean(), max=occ_thre)
-        ).reshape(self._binary.shape)
+        ).view(self._binary.shape)

    @torch.no_grad()
    def every_n_step(
@@ -289,7 +289,7 @@ class OccupancyGrid(Grid):
        return query_grid(
            samples,
            self._roi_aabb,
-            self.occs.reshape(self.resolution.tolist()),
+            self.binary,
            self.contraction_type,
        )