From 606bf42089b549a592c088836016ebd321e31f4c Mon Sep 17 00:00:00 2001 From: harisreedhar Date: Mon, 10 Mar 2025 13:49:50 +0530 Subject: [PATCH] fix --- face_swapper/config.ini | 1 + face_swapper/src/networks/masknet.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/face_swapper/config.ini b/face_swapper/config.ini index 022c809..881f72e 100644 --- a/face_swapper/config.ini +++ b/face_swapper/config.ini @@ -14,6 +14,7 @@ split_ratio = embedder_path = gazer_path = motion_extractor_path = +parser_path = [training.model.generator] identity_channels = diff --git a/face_swapper/src/networks/masknet.py b/face_swapper/src/networks/masknet.py index 859597f..7d26633 100644 --- a/face_swapper/src/networks/masknet.py +++ b/face_swapper/src/networks/masknet.py @@ -46,7 +46,7 @@ class MaskNet(nn.Module): output_tensor = up_sample(output_tensor) output_tensor = self.conv(output_tensor) - output_tensor = self.activation(output_tensor) + output_tensor = self.sigmoid(output_tensor) return output_tensor