fix bug in issue #88

OPEN-AIR-SUN · Oct 24, 2023 · 2666917 · 2666917
1 parent 9552167
commit 2666917
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/nsg/models/scene_graph.py b/nsg/models/scene_graph.py
@@ -692,8 +692,9 @@ def get_loss_dict(self, outputs, batch, metrics_dict=None):
             assert "depth_mask" in batch
             depth_gt = batch["depth_image"].to(self.device).float()
             depth_mask = batch["depth_mask"].to(self.device)
-            if not self.config.is_euclidean_depth:
-                depth_gt = depth_gt * outputs["directions_norm"]
+            # depth_gt multipled by directions_norm is excuted in the depth loss function
+            # if not self.config.is_euclidean_depth:
+            #     depth_gt = depth_gt * outputs["directions_norm"]
             depth_gt[~depth_mask] = 0.0  # to make it compatible with the automask of the depth loss
             predicted_depth = outputs["depth"].float()
             depth_loss = 0