Machine-Learning
/
AIIA
1
0
Fork 0
Code Issues 1 Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request 'updated version' (#22) from bug_fixes into main 

Reviewed-on: #22
This commit is contained in:
Falko Victor Habel 2025-03-03 17:24:58 +00:00
parent 3b018a6761 6221a90a56
commit f14ed76b42
4 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pyproject.toml
View File

@ -10,7 +10,7 @@ include = '\.pyi?$'
[project] [project]
name = "aiia" name = "aiia"
version = "0.1.4" version = "0.1.5"
description = "AIIA Deep Learning Model Implementation" description = "AIIA Deep Learning Model Implementation"
readme = "README.md" readme = "README.md"
authors = [ authors = [

2
setup.cfg
View File

@ -1,6 +1,6 @@
[metadata] [metadata]
name = aiia name = aiia
version = 0.1.4 version = 0.1.5
author = Falko Habel author = Falko Habel
author_email = falko.habel@gmx.de author_email = falko.habel@gmx.de
description = AIIA deep learning model implementation description = AIIA deep learning model implementation

2
src/aiia/__init__.py
View File

@ -4,4 +4,4 @@ from .data.DataLoader import DataLoader
from .pretrain.pretrainer import Pretrainer, ProjectionHead from .pretrain.pretrainer import Pretrainer, ProjectionHead
__version__ = "0.1.4" __version__ = "0.1.5"

8
src/aiia/data/DataLoader.py
View File

@ -177,7 +177,7 @@ class AIIADataset(torch.utils.data.Dataset):
self.items = items self.items = items
self.pretraining = pretraining self.pretraining = pretraining
self.transform = transforms.Compose([ self.transform = transforms.Compose([
transforms.Resize((384, 384)), transforms.Resize((352, 352)),
transforms.ToTensor() transforms.ToTensor()
]) ])
@ -193,7 +193,7 @@ class AIIADataset(torch.utils.data.Dataset):
raise ValueError(f"Invalid image at index {idx}") raise ValueError(f"Invalid image at index {idx}")
image = self.transform(image) 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}") raise ValueError(f"Invalid image shape at index {idx}: {image.shape}")
if task == 'denoise': if task == 'denoise':
@ -215,7 +215,7 @@ class AIIADataset(torch.utils.data.Dataset):
if not isinstance(image, Image.Image): if not isinstance(image, Image.Image):
raise ValueError(f"Invalid image at index {idx}") raise ValueError(f"Invalid image at index {idx}")
image = self.transform(image) 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}") raise ValueError(f"Invalid image shape at index {idx}: {image.shape}")
return image, label return image, label
else: else:
@ -223,6 +223,6 @@ class AIIADataset(torch.utils.data.Dataset):
image = self.transform(item) image = self.transform(item)
else: else:
image = self.transform(item[0]) 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}") raise ValueError(f"Invalid image shape at index {idx}: {image.shape}")
return image return image