From 41aab4ca0e6a9d0816e9c420a585e649e8892152 Mon Sep 17 00:00:00 2001 From: harisreedhar Date: Mon, 28 Oct 2024 22:02:44 +0530 Subject: [PATCH] changes --- facefusion/vision.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/facefusion/vision.py b/facefusion/vision.py index 6237a0b5..fa3f821b 100644 --- a/facefusion/vision.py +++ b/facefusion/vision.py @@ -250,12 +250,12 @@ 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: - sizes = numpy.linspace(16, target_vision_frame.shape[0], 3, endpoint = False, dtype = numpy.int32) + color_difference_sizes = numpy.linspace(16, target_vision_frame.shape[0], 3, endpoint = False, dtype = numpy.int32) target_vision_frame = target_vision_frame.astype(numpy.float32) / 255 source_vision_frame = source_vision_frame.astype(numpy.float32) / 255 - for size in sizes: - source_vision_frame = apply_color_difference(source_vision_frame, target_vision_frame, ( size, size )) + for color_difference_size in color_difference_sizes: + source_vision_frame = apply_color_difference(source_vision_frame, target_vision_frame, ( color_difference_size, color_difference_size )) target_vision_frame = apply_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