From d5232901fba21b122c0bbe3740476b59585541eb Mon Sep 17 00:00:00 2001 From: harisreedhar Date: Mon, 28 Oct 2024 22:33:05 +0530 Subject: [PATCH] further reduction --- facefusion/vision.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/facefusion/vision.py b/facefusion/vision.py index 39541222..587bfea7 100644 --- a/facefusion/vision.py +++ b/facefusion/vision.py @@ -250,21 +250,18 @@ def merge_tile_frames(tile_vision_frames : List[VisionFrame], temp_width : int, def match_frame_color(source_vision_frame : VisionFrame, target_vision_frame : VisionFrame) -> VisionFrame: - target_vision_frame = target_vision_frame.astype(numpy.float32) / 255 - source_vision_frame = source_vision_frame.astype(numpy.float32) / 255 color_difference_sizes = numpy.linspace(16, target_vision_frame.shape[0], 3, endpoint = False) for color_difference_size in color_difference_sizes: source_vision_frame = remove_color_difference(source_vision_frame, target_vision_frame, normalize_resolution(( color_difference_size, color_difference_size ))) target_vision_frame = remove_color_difference(source_vision_frame, target_vision_frame, target_vision_frame.shape[:2][::-1]) - target_vision_frame = numpy.multiply(target_vision_frame, 255).astype(numpy.uint8) return target_vision_frame def remove_color_difference(source_vision_frame : VisionFrame, target_vision_frame : VisionFrame, size : Size) -> VisionFrame: - source_frame_resize = cv2.resize(source_vision_frame, size, interpolation = cv2.INTER_AREA) - target_frame_resize = cv2.resize(target_vision_frame, size, interpolation = cv2.INTER_AREA) + source_frame_resize = cv2.resize(source_vision_frame, size, interpolation = cv2.INTER_AREA).astype(numpy.float32) + target_frame_resize = cv2.resize(target_vision_frame, size, interpolation = cv2.INTER_AREA).astype(numpy.float32) color_difference_frame = numpy.subtract(source_frame_resize, target_frame_resize) color_difference_frame = cv2.resize(color_difference_frame, target_vision_frame.shape[:2][::-1], interpolation = cv2.INTER_CUBIC) - target_vision_frame = numpy.add(target_vision_frame, color_difference_frame).clip(0, 1) + target_vision_frame = numpy.add(target_vision_frame, color_difference_frame).clip(0, 255).astype(numpy.uint8) return target_vision_frame