A.Cutout added in the training data augmentation

code/data.py

@@ -275,6 +275,7 @@ class TestDataModule(pl.LightningDataModule):
                 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),
                 ToTensor(),
             ],
             additional_targets={'pli_image': 'image', 'cyto_image': 'image'}