From 6221a90a56e2fe597349fccade86476313b71ec7 Mon Sep 17 00:00:00 2001
From: Falko Habel <falko.habel@gmx.de>
Date: Mon, 3 Mar 2025 18:24:32 +0100
Subject: [PATCH] updated version

---
 pyproject.toml              | 2 +-
 setup.cfg                   | 2 +-
 src/aiia/__init__.py        | 2 +-
 src/aiia/data/DataLoader.py | 8 ++++----
 4 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 3c4b7b6..62715b6 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -10,7 +10,7 @@ include = '\.pyi?$'
 
 [project]
 name = "aiia"
-version = "0.1.4"
+version = "0.1.5"
 description = "AIIA Deep Learning Model Implementation"
 readme = "README.md"
 authors = [
diff --git a/setup.cfg b/setup.cfg
index 75b1b49..0bd097e 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -1,6 +1,6 @@
 [metadata]
 name = aiia
-version = 0.1.4
+version = 0.1.5
 author = Falko Habel
 author_email = falko.habel@gmx.de
 description = AIIA deep learning model implementation
diff --git a/src/aiia/__init__.py b/src/aiia/__init__.py
index 5786564..19e0fe8 100644
--- a/src/aiia/__init__.py
+++ b/src/aiia/__init__.py
@@ -4,4 +4,4 @@ from .data.DataLoader import DataLoader
 from .pretrain.pretrainer import Pretrainer, ProjectionHead
 
 
-__version__ = "0.1.4"
+__version__ = "0.1.5"
diff --git a/src/aiia/data/DataLoader.py b/src/aiia/data/DataLoader.py
index b4cd944..8db0e8a 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((384, 384)),
+            transforms.Resize((352, 352)),
             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, 384, 384):
+            if image.shape != (3, 352, 352):
                 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, 384, 384):
+                if image.shape != (3, 352, 352):
                     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, 384, 384):
+                if image.shape != (3, 352, 352):
                     raise ValueError(f"Invalid image shape at index {idx}: {image.shape}")
                 return image