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