From c35b0a6f4cbd91ae207df32ee474095ea4bbd3bf Mon Sep 17 00:00:00 2001 From: harisreedhar Date: Mon, 3 Mar 2025 13:28:55 +0530 Subject: [PATCH] changes --- face_swapper/src/dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/face_swapper/src/dataset.py b/face_swapper/src/dataset.py index c4e7d57..2494e1c 100644 --- a/face_swapper/src/dataset.py +++ b/face_swapper/src/dataset.py @@ -36,7 +36,7 @@ class DynamicDataset(Dataset[Tensor]): transforms.ColorJitter(brightness = 0.2, contrast = 0.2, saturation = 0.2, hue = 0.1), transforms.RandomAffine(4, translate = (0.01, 0.01), scale = (0.98, 1.02), shear = (1, 1)), transforms.ToTensor(), - lambda temp: warp_tensor(temp.unsqueeze(0), 'vgg_face_hq_to_arcface_128_v2').squeeze(0), + transforms.Lambda(lambda temp: warp_tensor(temp.unsqueeze(0), 'vgg_face_hq_to_arcface_128_v2').squeeze(0)), transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)) ])