diff --git a/nerfstudio/models/base_surface_model.py b/nerfstudio/models/base_surface_model.py index 5d7fff11..f8680249 100644 --- 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)