From 13548173044176570946765c6b18fdaf0ae8240f Mon Sep 17 00:00:00 2001 From: Falko Habel Date: Mon, 3 Mar 2025 18:11:29 +0100 Subject: [PATCH] to 384 from 400 --- src/aiia/data/DataLoader.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/src/aiia/data/DataLoader.py b/src/aiia/data/DataLoader.py index 89ba400..b4cd944 100644 --- a/src/aiia/data/DataLoader.py +++ b/src/aiia/data/DataLoader.py @@ -177,7 +177,7 @@ class AIIADataset(torch.utils.data.Dataset): self.items = items self.pretraining = pretraining self.transform = transforms.Compose([ - transforms.Resize((400, 400)), + transforms.Resize((384, 384)), transforms.ToTensor() ]) @@ -193,7 +193,7 @@ class AIIADataset(torch.utils.data.Dataset): raise ValueError(f"Invalid image at index {idx}") image = self.transform(image) - if image.shape != (3, 400, 400): + if image.shape != (3, 384, 384): raise ValueError(f"Invalid image shape at index {idx}: {image.shape}") if task == 'denoise': @@ -215,7 +215,7 @@ class AIIADataset(torch.utils.data.Dataset): if not isinstance(image, Image.Image): raise ValueError(f"Invalid image at index {idx}") image = self.transform(image) - if image.shape != (3, 400, 400): + if image.shape != (3, 384, 384): raise ValueError(f"Invalid image shape at index {idx}: {image.shape}") return image, label else: @@ -223,6 +223,6 @@ class AIIADataset(torch.utils.data.Dataset): image = self.transform(item) else: image = self.transform(item[0]) - if image.shape != (3, 400, 400): + if image.shape != (3, 384, 384): raise ValueError(f"Invalid image shape at index {idx}: {image.shape}") return image