Adjusting Instant NGP to Inner AABB

Anthony-Tafoya · Anthony-Tafoya · commit a95214292649 · 2024-09-19T22:10:33.000Z
diff --git a/nerfstudio/models/instant_ngp.py b/nerfstudio/models/instant_ngp.py
@@ -118,7 +118,7 @@ def populate_modules(self):
             self.config.render_step_size = ((self.scene_aabb[3:] - self.scene_aabb[:3]) ** 2).sum().sqrt().item() / 1000
         # Occupancy Grid.
         self.occupancy_grid = nerfacc.OccGridEstimator(
-            roi_aabb=self.scene_aabb,
+            roi_aabb=self.scene_aabb * 2 ** -(levels - 1),
             resolution=self.config.grid_resolution,
             levels=self.config.grid_levels,
         )

Original file line number	Diff line number	Diff line change
`@@ -118,7 +118,7 @@ def populate_modules(self):`
`118`	`118`	`self.config.render_step_size = ((self.scene_aabb[3:] - self.scene_aabb[:3]) ** 2).sum().sqrt().item() / 1000`
`119`	`119`	`# Occupancy Grid.`
`120`	`120`	`self.occupancy_grid = nerfacc.OccGridEstimator(`
`121`		`- roi_aabb=self.scene_aabb,`
	`121`	`+ roi_aabb=self.scene_aabb * 2 ** -(levels - 1),`
`122`	`122`	`resolution=self.config.grid_resolution,`
`123`	`123`	`levels=self.config.grid_levels,`
`124`	`124`	`)`