diff --git a/facefusion/face_helper.py b/facefusion/face_helper.py index a6900d3..cc37fe0 100644 --- a/facefusion/face_helper.py +++ b/facefusion/face_helper.py @@ -106,6 +106,7 @@ def paste_back(temp_vision_frame : VisionFrame, crop_vision_frame : VisionFrame, inverse_mask = cv2.warpAffine(crop_mask, paste_matrix, (paste_width, paste_height)).clip(0, 1) inverse_mask = numpy.expand_dims(inverse_mask, axis = -1) inverse_vision_frame = cv2.warpAffine(crop_vision_frame, paste_matrix, (paste_width, paste_height), borderMode = cv2.BORDER_REPLICATE) + temp_vision_frame = temp_vision_frame.copy() paste_vision_frame = temp_vision_frame[y_min:y_max, x_min:x_max] paste_vision_frame = paste_vision_frame * (1 - inverse_mask) + inverse_vision_frame * inverse_mask temp_vision_frame[y_min:y_max, x_min:x_max] = paste_vision_frame.astype(temp_vision_frame.dtype)