From ceb3c0cfdfcc38e4bea1102bd5badb443c197c33 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Mon, 3 Mar 2025 18:29:51 +0100 Subject: [PATCH] Rename to AugmentTransform --- face_swapper/src/dataset.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/face_swapper/src/dataset.py b/face_swapper/src/dataset.py index 95e6304..a0f00e7 100644 --- a/face_swapper/src/dataset.py +++ b/face_swapper/src/dataset.py @@ -38,7 +38,7 @@ class DynamicDataset(Dataset[Tensor]): [ transforms.ToPILImage(), transforms.Resize((256, 256), interpolation = transforms.InterpolationMode.BICUBIC), - AlterTransform(), + AugmentTransform(), transforms.ToTensor(), WarpTransform(self.warp_template), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)) @@ -68,7 +68,7 @@ class DynamicDataset(Dataset[Tensor]): return source_tensor, target_tensor -class AlterTransform: +class AugmentTransform: def __init__(self) -> None: self.transforms = self.compose_transforms()