Add HorizontalFlip to AugmentTransform

face_swapper/README.md

@@ -1,7 +1,7 @@
 Face Swapper
 ============
 
-> Face shape and feature aware identity transfer.
+> Face shape and occlusion aware identity transfer.
 
 ![License](https://img.shields.io/badge/license-ResearchRAIL--MS-red)
 

face_swapper/src/dataset.py

@@ -84,12 +84,13 @@ class AugmentTransform:
 	def compose_transforms() -> albumentations.Compose:
 		return albumentations.Compose(
 		[
-			albumentations.RandomBrightnessContrast(p = 0.3),
+			albumentations.HorizontalFlip(),
 			albumentations.OneOf(
 			[
 				albumentations.MotionBlur(p = 0.1),
 				albumentations.MedianBlur(p = 0.1)
 			], p = 0.3),
+			albumentations.RandomBrightnessContrast(p = 0.3),
 			albumentations.ColorJitter(p = 0.1)
 		])