Merge pull request #7 (Augmentation bug fix) from olivesgatech/develop

Augmentation bug fix
yalaudah · Apr 3, 2019 · 0d4ff25 · 0d4ff25
2 parents 2f6d85f + 01e10c0
commit 0d4ff25
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/core/augmentations.py b/core/augmentations.py
@@ -68,13 +68,15 @@ def __call__(self, img, mask):
 class RandomHorizontallyFlip(object):
     def __call__(self, img, mask):
         if random.random() < 0.5:
-            return img.transpose(Image.FLIP_LEFT_RIGHT), mask.transpose(Image.FLIP_LEFT_RIGHT)
+            #Note: we use FLIP_TOP_BOTTOM here intentionaly. Due to the dimensions of the image,
+            # it ends up being a horizontal flip.
+            return img.transpose(Image.FLIP_TOP_BOTTOM), mask.transpose(Image.FLIP_TOP_BOTTOM)
         return img, mask
 
 class RandomVerticallyFlip(object):
     def __call__(self, img, mask):
         if random.random() < 0.5:
-            return img.transpose(Image.FLIP_TOP_BOTTOM), mask.transpose(Image.FLIP_TOP_BOTTOM)
+            return img.transpose(Image.FLIP_LEFT_RIGHT), mask.transpose(Image.FLIP_LEFT_RIGHT)
         return img, mask
 
 class FreeScale(object):

diff --git a/patch_train.py b/patch_train.py
@@ -96,7 +96,7 @@ def train(args):
     # Setup Augmentations
     if args.aug:
         data_aug = Compose(
-            [RandomRotate(30), RandomHorizontallyFlip(), AddNoise()])
+            [RandomRotate(10), RandomHorizontallyFlip(), AddNoise()])
     else:
         data_aug = None
 

diff --git a/section_train.py b/section_train.py
@@ -70,7 +70,7 @@ def train(args):
     # Setup Augmentations
     if args.aug:
         data_aug = Compose(
-            [RandomRotate(25), RandomHorizontallyFlip(), AddNoise()])
+            [RandomRotate(10), RandomHorizontallyFlip(), AddNoise()])
     else:
         data_aug = None