From 766e74a6a2c3d8d09f4023ed6de422bba8b9b642 Mon Sep 17 00:00:00 2001 From: Bingxin Date: Fri, 16 Aug 2024 11:48:53 +0200 Subject: [PATCH] [FIX] validation image shape error #100 --- src/trainer/marigold_trainer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/trainer/marigold_trainer.py b/src/trainer/marigold_trainer.py index a1596f8..8be2dee 100644 --- 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()