From 2845c78849b1332333e5150dea2d2041db1d45c5 Mon Sep 17 00:00:00 2001
From: Schneider Leo <leo.schneider@etu.ec-lyon.fr>
Date: Fri, 4 Apr 2025 16:57:34 +0200
Subject: [PATCH] revert : old transform version
---
image_ref/dataset_ref.py | 2 ++
1 file changed, 2 insertions(+)
diff --git a/image_ref/dataset_ref.py b/image_ref/dataset_ref.py
index e4014100..3472d627 100644
--- a/image_ref/dataset_ref.py
+++ b/image_ref/dataset_ref.py
@@ -170,6 +170,8 @@ def load_data_duo(base_dir_train, base_dir_test, batch_size, shuffle=True, noise
ref_transform = transforms.Compose(
[transforms.Resize((224, 224)),
+ Threshold_noise(noise_threshold),
+ Log_normalisation(),
transforms.Normalize(0.5, 0.5)])
print('Default val transform')
--
GitLab