From 95a63ea7a250d3dfd9a389916780c856263b55a4 Mon Sep 17 00:00:00 2001 From: harisreedhar Date: Fri, 1 Nov 2024 18:11:25 +0530 Subject: [PATCH] add both mask instead of multiply --- facefusion/processors/modules/deep_swapper.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/facefusion/processors/modules/deep_swapper.py b/facefusion/processors/modules/deep_swapper.py index 0d34d2b7..0f7c2370 100755 --- a/facefusion/processors/modules/deep_swapper.py +++ b/facefusion/processors/modules/deep_swapper.py @@ -128,8 +128,7 @@ def swap_face(target_face : Face, temp_vision_frame : VisionFrame) -> VisionFram crop_vision_frame, crop_source_mask, crop_target_mask = forward(crop_vision_frame) crop_vision_frame = normalize_crop_frame(crop_vision_frame) crop_vision_frame = match_frame_color_with_mask(crop_vision_frame_raw, crop_vision_frame, crop_source_mask, crop_target_mask) - crop_masks.append(feather_crop_mask(crop_source_mask)) - crop_masks.append(feather_crop_mask(crop_target_mask)) + crop_masks.append(numpy.maximum.reduce([ feather_crop_mask(crop_source_mask), feather_crop_mask(crop_target_mask) ]).clip(0, 1)) crop_mask = numpy.minimum.reduce(crop_masks).clip(0, 1) paste_vision_frame = paste_back(temp_vision_frame, crop_vision_frame, crop_mask, affine_matrix) return paste_vision_frame