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

Compare commits

base: Machine-Learning:d41ec2d82a5959c4225fe9e7ab96cd3a2aca1491
Branches Tags
Machine-Learning:main
..
compare: Machine-Learning:8703835ea78a6e18e6fe9851a2e34cc2663491e2
Branches Tags
Machine-Learning:main

No commits in common. "d41ec2d82a5959c4225fe9e7ab96cd3a2aca1491" and "8703835ea78a6e18e6fe9851a2e34cc2663491e2" have entirely different histories.
d41ec2d82a ... 8703835ea7

1 changed files with 4 additions and 4 deletions
Show Stats Expand all files Collapse all files

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

@ -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((410, 410)),
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, 410, 410):
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, 410, 410):
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, 410, 410):
raise ValueError(f"Invalid image shape at index {idx}: {image.shape}")
return image