[FIX] validation image shape error #100

prs-eth · Aug 16, 2024 · 766e74a · 766e74a
1 parent 10e775f
commit 766e74a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/trainer/marigold_trainer.py b/src/trainer/marigold_trainer.py
@@ -1,5 +1,5 @@
 # An official reimplemented version of Marigold training script.
-# Last modified: 2024-04-29
+# Last modified: 2024-08-16
 #
 # Copyright 2023 Bingxin Ke, ETH Zurich. All rights reserved.
 #
@@ -507,7 +507,7 @@ def validate_single_dataset(
         ):
             assert 1 == data_loader.batch_size
             # Read input image
-            rgb_int = batch["rgb_int"].squeeze()  # [3, H, W]
+            rgb_int = batch["rgb_int"]  # [B, 3, H, W]
             # GT depth
             depth_raw_ts = batch["depth_raw_linear"].squeeze()
             depth_raw = depth_raw_ts.numpy()