From 23ab7dc89ded8060c8c44dcfbbf54eb2a60265e5 Mon Sep 17 00:00:00 2001 From: harisreedhar Date: Mon, 3 Mar 2025 13:20:37 +0530 Subject: [PATCH] fix --- 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 8fa31d0..c4e7d57 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), + 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)) ])