CalvinBackup/facefusion-labs
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-05-22 23:59:40 +02:00
Code Issues Packages Projects Releases Wiki Activity

more augmentation

Browse Source
This commit is contained in:
harisreedhar
2025-04-10 11:33:21 +05:30
parent dafada11bc
commit 056bacb7de
1 changed files with 7 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
face_swapper/src/dataset.py
+7 -4
View File
@@ -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)
])