From ead12e3fb23514d65b97c8dcb2eca60b50477c06 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Sat, 15 Jul 2023 18:13:10 +0200 Subject: [PATCH] Temp fix for enhancer --- roop/processors/frame/face_enhancer.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/roop/processors/frame/face_enhancer.py b/roop/processors/frame/face_enhancer.py index 63d32a41a..a4c9a3a07 100644 --- a/roop/processors/frame/face_enhancer.py +++ b/roop/processors/frame/face_enhancer.py @@ -67,7 +67,7 @@ def enhance_face(temp_frame: Frame) -> Frame: return temp_frame -def process_frame(source_face: Face, temp_frame: Frame) -> Frame: +def process_frame(source_face: Face, reference_face: Face, temp_frame: Frame) -> Frame: target_face = get_one_face(temp_frame) if target_face: temp_frame = enhance_face(temp_frame) @@ -77,7 +77,7 @@ def process_frame(source_face: Face, temp_frame: Frame) -> Frame: def process_frames(source_path: str, temp_frame_paths: List[str], update: Callable[[], None]) -> None: for temp_frame_path in temp_frame_paths: temp_frame = cv2.imread(temp_frame_path) - result = process_frame(None, temp_frame) + result = process_frame(None, None, temp_frame) cv2.imwrite(temp_frame_path, result) if update: update()