From e5ed04fa25f83644c48a9850a4aca2a0c3aa8e8b Mon Sep 17 00:00:00 2001 From: Falko Habel Date: Mon, 3 Mar 2025 18:31:26 +0100 Subject: [PATCH] downsized transfomrers per iamge --- 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 8db0e8a..20a9f5d 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((352, 352)), + transforms.Resize((256, 256)), 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, 352, 352): + if image.shape != (3, 256, 256): 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, 352, 352): + if image.shape != (3, 256, 256): 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, 352, 352): + if image.shape != (3, 256, 256): raise ValueError(f"Invalid image shape at index {idx}: {image.shape}") return image