fix bug in #89

OPEN-AIR-SUN · Oct 27, 2023 · 99e4a03 · 99e4a03
1 parent 2666917
commit 99e4a03
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nsg/models/scene_graph.py b/nsg/models/scene_graph.py
@@ -395,7 +395,7 @@ def get_outputs(self, ray_bundle: RayBundle):
             # (n_rays, n_samples)
 
             output_bg_insec_density = output_bg_density[intersection_map[..., 0]]  # (n_intersects, n_samples)
-            mask = (bg_samples_z_vals > z_vals_in_o.unsqueeze(-1)) & (bg_samples_z_vals < z_vals_out_o.unsqueeze(-1))
+            mask = (bg_samples_z_vals > z_vals_in_w.unsqueeze(-1)) & (bg_samples_z_vals < z_vals_out_w.unsqueeze(-1))
             output_bg_insec_density[mask] = 0.0
 
         track_idx = obj_pose[..., 4]  # (n_intersects, )