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

further reduction

Browse Source
This commit is contained in:
harisreedhar 2024-10-28 22:33:05 +05:30
parent 9388073d67
commit d5232901fb
1 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
facefusion/vision.py
View File

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