From 5f123ed968ab930e030799ce75a7d7c7ce8dae98 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Tue, 1 Apr 2025 14:59:43 +0200
Subject: [PATCH] fix : dataloader batched
---
image_ref/dataset_ref.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/image_ref/dataset_ref.py b/image_ref/dataset_ref.py
index 5dd8ff30..738820aa 100644
--- a/image_ref/dataset_ref.py
+++ b/image_ref/dataset_ref.py
@@ -248,8 +248,8 @@ def load_data_duo_batched(base_dir, shuffle=True, noise_threshold=0, ref_dir = N
transforms.Normalize(0.5, 0.5)])
print('Default val transform')
- train_dataset = ImageFolderDuo(root=base_dir, transform=train_transform, ref_dir = ref_dir)
- val_dataset = ImageFolderDuo(root=base_dir, transform=val_transform, ref_dir = ref_dir)
+ train_dataset = ImageFolderDuo_Batched(root=base_dir, transform=train_transform, ref_dir = ref_dir)
+ val_dataset = ImageFolderDuo_Batched(root=base_dir, transform=val_transform, ref_dir = ref_dir)
generator1 = torch.Generator().manual_seed(42)
indices = torch.randperm(len(train_dataset), generator=generator1)
val_size = len(train_dataset) // 5
--
GitLab