Disable ghosting augmentation to check motion in isolation

dzenanz · Dec 14, 2021 · 61ceb84 · 61ceb84
1 parent 494ca43
commit 61ceb84
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/miqa/learning/nn_training.py b/miqa/learning/nn_training.py
@@ -306,7 +306,7 @@ def create_train_and_test_data_loaders(df, count_train):
     ghosting = CustomGhosting(p=0.3, intensity=(0.2, 0.8))
     motion = CustomMotion(p=0.2, degrees=5.0, translation=5.0, num_transforms=1)
     # transforms = torchio.Compose([rescale, ghosting, motion])
-    transforms = torchio.Compose([rescale, ghosting])
+    transforms = torchio.Compose([rescale, motion])
 
     # create a training data loader
     train_ds = torchio.SubjectsDataset(train_files, transform=transforms)