From 056bacb7de32fe8a282457f994ed42c7d75b0cd1 Mon Sep 17 00:00:00 2001 From: harisreedhar Date: Thu, 10 Apr 2025 11:33:21 +0530 Subject: [PATCH] more augmentation --- face_swapper/src/dataset.py | 11 +++++++---- 1 file changed, 7 insertions(+), 4 deletions(-) diff --git a/face_swapper/src/dataset.py b/face_swapper/src/dataset.py index 183bb41..0a14eb4 100644 --- a/face_swapper/src/dataset.py +++ b/face_swapper/src/dataset.py @@ -88,10 +88,13 @@ class AugmentTransform: 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) + albumentations.ZoomBlur(p = 0.1) + ], p = 0.2), + albumentations.RandomBrightnessContrast(p = 0.7), + albumentations.ColorJitter(p = 0.2), + albumentations.RGBShift(p = 0.7), + albumentations.Illumination(p = 0.2), + albumentations.Affine(translate_percent = (-0.03, 0.03), scale = (0.98, 1.02), rotate = (-2, 2), border_mode = 1, p = 0.7) ])