diff --git a/image_ref/dataset_ref.py b/image_ref/dataset_ref.py
index e4014100ec99d2a808c98bee4b03f02e67155cf4..3472d627e5a61a4c7d5f2a7f0768486c80081375 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')