From 8474c413c1552154f8565a21920f23f48355ee97 Mon Sep 17 00:00:00 2001
From: "s.islam" <s.islam@fz-juelich.de>
Date: Fri, 29 Apr 2022 15:25:49 +0200
Subject: [PATCH] Using 2 channels(NTransmittance & Retardation) for more
regional information with sigmoid at decoder end
---
code/data.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/code/data.py b/code/data.py
index 3dc5ced..da228d7 100644
--- a/code/data.py
+++ b/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'}
--
GitLab