From 1447eb45e5da947ee191c354f98cffd6272eeaa6 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Mon, 23 Jun 2025 10:30:07 +0200
Subject: [PATCH] fiw : data_loader
---
dataset/dataset.py | 8 ++------
1 file changed, 2 insertions(+), 6 deletions(-)
diff --git a/dataset/dataset.py b/dataset/dataset.py
index 6130f5d..b78db35 100644
--- a/dataset/dataset.py
+++ b/dataset/dataset.py
@@ -252,17 +252,13 @@ def load_data_duo(base_dir_train, base_dir_val, base_dir_test, batch_size, shuff
[transforms.Resize((224, 224)),
Threshold_noise(noise_threshold),
- Log_normalisation(),
- Repeat(),
- transforms.Normalize(mean=[0.485, 0.456, 0.406],std=[0.229, 0.224, 0.225])])
+ Log_normalisation(),])
print('Default train transform')
val_transform = transforms.Compose(
[transforms.Resize((224, 224)),
Threshold_noise(noise_threshold),
- Log_normalisation(),
- Repeat(),
- transforms.Normalize(mean=[0.485, 0.456, 0.406],std=[0.229, 0.224, 0.225])])
+ Log_normalisation(),])
print('Default val transform')
--
GitLab