From fea75ff9493f0f4fc89ffd6760ffc3ad7627b760 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Tue, 4 Mar 2025 12:17:59 +0100 Subject: [PATCH] Remove useless comma --- face_swapper/src/dataset.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/face_swapper/src/dataset.py b/face_swapper/src/dataset.py index 852648b..7464fb5 100644 --- a/face_swapper/src/dataset.py +++ b/face_swapper/src/dataset.py @@ -86,7 +86,7 @@ class AugmentTransform: albumentations.MotionBlur(p = 0.1), albumentations.MedianBlur(p = 0.1) ], p = 0.3), - albumentations.ColorJitter(p = 0.1), + albumentations.ColorJitter(p = 0.1) ]) @@ -97,4 +97,3 @@ class WarpTransform: def __call__(self, input_tensor : Tensor) -> Tensor: temp_tensor = input_tensor.unsqueeze(0) return warp_tensor(temp_tensor, self.warp_template).squeeze(0) -