switch normal/depth gt for image metrics to be consistent with rgb (z…

…ju3dv#6)
siatheindochinese · Dec 24, 2022 · bf249d6 · bf249d6
1 parent ee2f170
commit bf249d6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nerfstudio/models/base_surface_model.py b/nerfstudio/models/base_surface_model.py
@@ -417,7 +417,7 @@ def get_image_metrics_and_images(
             )
             depth_pred = depth_pred * scale + shift
 
-            combined_depth = torch.cat([depth_pred, depth_gt[..., None]], dim=1)
+            combined_depth = torch.cat([depth_gt[..., None]], depth_pred, dim=1)
             combined_depth = colormaps.apply_depth_colormap(combined_depth)
         else:
             depth = colormaps.apply_depth_colormap(
@@ -428,7 +428,7 @@ def get_image_metrics_and_images(
 
         if "normal" in batch:
             normal_gt = (batch["normal"].to(self.device) + 1.0) / 2.0
-            combined_normal = torch.cat([normal, normal_gt], dim=1)
+            combined_normal = torch.cat([normal_gt, normal], dim=1)
         else:
             combined_normal = torch.cat([normal], dim=1)