mirrors/facefusion
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

changes

Browse Source
This commit is contained in:
harisreedhar 2024-10-28 22:02:44 +05:30
parent bede313c45
commit 41aab4ca0e
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
facefusion/vision.py
View File

@ -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