From 7cf5609c1f79dd5f1192358cca9d0667bb40d771 Mon Sep 17 00:00:00 2001 From: harisreedhar Date: Wed, 5 Mar 2025 22:21:26 +0530 Subject: [PATCH] changes --- face_swapper/README.md | 1 - face_swapper/config.ini | 1 - face_swapper/src/models/loss.py | 10 +++------- 3 files changed, 3 insertions(+), 9 deletions(-) diff --git a/face_swapper/README.md b/face_swapper/README.md index 65d47ed..7772c88 100644 --- a/face_swapper/README.md +++ b/face_swapper/README.md @@ -72,7 +72,6 @@ attribute_weight = 10.0 reconstruction_weight = 20.0 identity_weight = 20.0 gaze_weight = 0.0 -gaze_scale_factor = 1.0 pose_weight = 0.0 expression_weight = 0.0 ``` diff --git a/face_swapper/config.ini b/face_swapper/config.ini index ad69109..0f3c702 100644 --- a/face_swapper/config.ini +++ b/face_swapper/config.ini @@ -34,7 +34,6 @@ attribute_weight = reconstruction_weight = identity_weight = gaze_weight = -gaze_scale_factor = pose_weight = expression_weight = diff --git a/face_swapper/src/models/loss.py b/face_swapper/src/models/loss.py index 6272ba9..bf6ba01 100644 --- a/face_swapper/src/models/loss.py +++ b/face_swapper/src/models/loss.py @@ -169,13 +169,9 @@ class GazeLoss(nn.Module): return gaze_loss, weighted_gaze_loss def detect_gaze(self, input_tensor : Tensor) -> Gaze: - scale_factor = CONFIG.getfloat('training.losses', 'gaze_scale_factor') - y_min = int(60 * scale_factor) - y_max = int(224 * scale_factor) - x_min = int(16 * scale_factor) - x_max = int(205 * scale_factor) - - crop_tensor = input_tensor[:, :, y_min:y_max, x_min:x_max] + transform_size = CONFIG.getint('training.dataset', 'transform_size') + crop_sizes = (torch.tensor([ 0.235, 0.875, 0.0625, 0.8 ]) * transform_size).int() + crop_tensor = input_tensor[:, :, crop_sizes[0]:crop_sizes[1], crop_sizes[2]:crop_sizes[3]] crop_tensor = (crop_tensor + 1) * 0.5 crop_tensor = transforms.Normalize(mean = [ 0.485, 0.456, 0.406 ], std = [ 0.229, 0.224, 0.225 ])(crop_tensor) crop_tensor = nn.functional.interpolate(crop_tensor, size = 448, mode = 'bicubic')