Skip to content

GitLab

Unverified Commit fe75cb86 authored by Thomas Wang's avatar Thomas Wang Committed by GitHub
Browse files

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

parent 0e1e7cb7
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
nerfacc/grid.py
View file @ fe75cb86
...@@ -236,7 +236,7 @@ class OccupancyGrid(Grid): ...@@ -236,7 +236,7 @@ class OccupancyGrid(Grid):
# ) # )
self._binary = ( self._binary = (
self.occs > torch.clamp(self.occs.mean(), max=occ_thre) self.occs > torch.clamp(self.occs.mean(), max=occ_thre)
).reshape(self._binary.shape) ).view(self._binary.shape)
@torch.no_grad() @torch.no_grad()
def every_n_step( def every_n_step(
...@@ -289,7 +289,7 @@ class OccupancyGrid(Grid): ...@@ -289,7 +289,7 @@ class OccupancyGrid(Grid):
return query_grid( return query_grid(
samples, samples,
self._roi_aabb, self._roi_aabb,
self.occs.reshape(self.resolution.tolist()), self.binary,
self.contraction_type, self.contraction_type,
) )
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment