Using 2 channels(NTransmittance & Retardation) for more regional information...

Using 2 channels(NTransmittance & Retardation) for more regional information with sigmoid at decoder end

code/data.py

@@ -294,10 +294,10 @@ class TestDataModule(pl.LightningDataModule):
         # Augmentations for training
         self._train_transforms = A.Compose(
             [
-                A.Affine(rotate=(-180, 180), translate_percent=(0.1, 0.1), shear=(-30, 30), cval=0, p=0.9),
+                A.Affine(rotate=(-180, 180), translate_percent=(0.1, 0.1), shear=(-30, 30), cval=0, p=1),
                 A.CenterCrop(p=1, height=self.patch_size, width=self.patch_size),
-                A.HorizontalFlip(p=0.9),
-                A.VerticalFlip(p=0.9),
+                A.HorizontalFlip(p=0.5),
+                A.VerticalFlip(p=0.5),
                 ToTensor(),
             ],
             additional_targets={'pli_image': 'image', 'cyto_image': 'image'}