From bab666d6f9216a9f24faa84ead2d006b76f30159 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Mon, 30 Oct 2023 22:45:10 +0100 Subject: [PATCH] Use cv2.blur over cv2.GaussianBlur for better performance --- facefusion/face_helper.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/facefusion/face_helper.py b/facefusion/face_helper.py index e29ab5d8..e3baad76 100644 --- a/facefusion/face_helper.py +++ b/facefusion/face_helper.py @@ -44,9 +44,9 @@ def paste_back(temp_frame: Frame, crop_frame: Frame, affine_matrix: Matrix) -> F inverse_mask_size = int(numpy.sqrt(numpy.sum(inverse_crop_frame == 255))) kernel_size = max(inverse_mask_size // 10, 10) inverse_crop_frame = cv2.erode(inverse_crop_frame, numpy.ones((kernel_size, kernel_size))) - kernel_size = max(inverse_mask_size // 20, 5) + kernel_size = max(inverse_mask_size // 30, 5) blur_size = kernel_size * 2 + 1 - inverse_blur_frame = cv2.GaussianBlur(inverse_crop_frame, (blur_size , blur_size), 0) / 255 + inverse_blur_frame = cv2.blur(inverse_crop_frame, (blur_size , blur_size)) / 255 inverse_blur_frame = numpy.reshape(inverse_blur_frame, [ temp_frame_height, temp_frame_width, 1 ]) temp_frame = inverse_blur_frame * inverse_temp_frame + (1 - inverse_blur_frame) * temp_frame temp_frame = temp_frame.astype(numpy.uint8)