diff --git a/dataset/dataset.py b/dataset/dataset.py index 6130f5db9f96cf62bbfe45b95ae6b07bb482828c..b78db350f9c5088e3c9ab38f84a5b5a89e582348 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')