CalvinBackup/facefusion-labs
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion-labs.git synced 2026-04-19 15:56:37 +02:00
Code Issues Packages Projects Releases Wiki Activity

Some polishing on augmentation (#82)

Browse Source
This commit is contained in:
Henry Ruhs
2025-06-09 07:49:00 +02:00
committed by GitHub
parent 143b594ee6
commit 3e9c8a37e7
1 changed files with 22 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
hyperswap/src/dataset.py
+22 -8
View File
@@ -146,15 +146,29 @@ class AugmentTransform:
def compose_transforms() -> albumentations.Compose:
return albumentations.Compose(
[
albumentations.HorizontalFlip(),
albumentations.HorizontalFlip(p = 0.5),
albumentations.OneOf(
[
albumentations.MotionBlur(p = 0.1),
albumentations.ZoomBlur(max_factor = (1.0, 1.1), p = 0.1)
albumentations.MotionBlur(),
albumentations.ZoomBlur(max_factor = (1.0, 1.2))
], p = 0.1),
albumentations.OneOf(
[
albumentations.RandomGamma(),
albumentations.RandomBrightnessContrast(),
albumentations.Illumination()
], 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.3)
albumentations.OneOf(
[
albumentations.ColorJitter(),
albumentations.RGBShift(),
albumentations.HueSaturationValue()
], p = 0.2),
albumentations.Affine(
translate_percent = (-0.05, 0.05),
scale = (0.95, 1.05),
rotate = (-2, 2),
border_mode = 1,
p = 0.2
)
])