From 8f76d96bb2d4b8f39067a6b106e27a2093b68493 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Tue, 11 Mar 2025 13:17:01 +0100 Subject: [PATCH] Name color tensor -> overlay tensor --- face_swapper/src/helper.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/face_swapper/src/helper.py b/face_swapper/src/helper.py index 0e223bc..352cfd4 100644 --- a/face_swapper/src/helper.py +++ b/face_swapper/src/helper.py @@ -39,8 +39,8 @@ def calc_embedding(embedder : EmbedderModule, input_tensor : Tensor, padding : P def overlay_mask(target_tensor : Tensor, mask_tensor : Tensor) -> Tensor: - temp_tensor = torch.zeros(*target_tensor.shape, dtype = target_tensor.dtype, device = target_tensor.device) - temp_tensor[:, 2, :, :] = 1 + overlay_tensor = torch.zeros(*target_tensor.shape, dtype = target_tensor.dtype, device = target_tensor.device) + overlay_tensor[:, 2, :, :] = 1 mask_tensor = mask_tensor.repeat(1, 3, 1, 1).clamp(0, 0.8) - output_tensor = target_tensor * (1 - mask_tensor) + temp_tensor * mask_tensor + output_tensor = target_tensor * (1 - mask_tensor) + overlay_tensor * mask_tensor return output_tensor