A.Cutout and A.Blur removed, More A.Affine transform added.

code/data.py

@@ -123,12 +123,12 @@ class TestDataModule(pl.LightningDataModule):
 
             #For single channel
             #For JSC Training.
-            pli_path = '/p/fastdata/pli/Private/oberstrass1/datasets/vervet1818/vervet1818-stained/data/aligned/pli/NTransmittance'
-            cyto_path = '/p/fastdata/pli/Private/oberstrass1/datasets/vervet1818/vervet1818-stained/data/aligned/stained'
+            #pli_path = '/p/fastdata/pli/Private/oberstrass1/datasets/vervet1818/vervet1818-stained/data/aligned/pli/NTransmittance'
+            #cyto_path = '/p/fastdata/pli/Private/oberstrass1/datasets/vervet1818/vervet1818-stained/data/aligned/stained'
 
             #For Local Machine Training.
-            #pli_path = '/media/tushar/A2246889246861F1/Master Thesis MAIA/example-data/pli/NTransmittance'
-            #cyto_path = '/media/tushar/A2246889246861F1/Master Thesis MAIA/example-data/stained'
+            pli_path = '/media/tushar/A2246889246861F1/Master Thesis MAIA/example-data/pli/NTransmittance'
+            cyto_path = '/media/tushar/A2246889246861F1/Master Thesis MAIA/example-data/stained'
             
             pli_files_list = [file for file in os.listdir(pli_path) if
                               file.endswith(('.h5', '.hdf', '.h4', '.hdf4', '.he2', '.hdf5', '.he5'))]
@@ -270,12 +270,10 @@ class TestDataModule(pl.LightningDataModule):
         # Augmentations for training
         self._train_transforms = A.Compose(
             [
-                A.Affine(rotate=(-180, 180)),
+                A.Affine(rotate=(-180, 180), translate_percent=(0.1, 0.1), shear=(-30, 30), cval=0, p=0.9),
                 A.CenterCrop(p=1, height=self.patch_size, width=self.patch_size),
-                A.HorizontalFlip(p=0.5),
-                A.VerticalFlip(p=0.5),
-                A.Blur(blur_limit=3, p=1),
-                A.Cutout(num_holes=102, max_h_size=8, max_w_size=8, fill_value=0, p=0.9),
+                A.HorizontalFlip(p=0.9),
+                A.VerticalFlip(p=0.9),
                 ToTensor(),
             ],
             additional_targets={'pli_image': 'image', 'cyto_image': 'image'}