From 0743b99347aad6dd6b1457fb0135d48f53fe31b1 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Sun, 23 Mar 2025 21:36:03 +0100 Subject: [PATCH] Add HorizontalFlip to AugmentTransform --- face_swapper/README.md | 2 +- face_swapper/src/dataset.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/face_swapper/README.md b/face_swapper/README.md index 13889e6..7fd5f9e 100644 --- a/face_swapper/README.md +++ b/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) diff --git a/face_swapper/src/dataset.py b/face_swapper/src/dataset.py index ebf7270..183bb41 100644 --- a/face_swapper/src/dataset.py +++ b/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) ])