From d3f38c7524953eb8f7e01ef167fdaa5439107504 Mon Sep 17 00:00:00 2001 From: K1llMan Date: Thu, 15 Jun 2023 20:58:54 +0300 Subject: [PATCH] Fix preview (#499) --- roop/ui.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/roop/ui.py b/roop/ui.py index 44b2fd5..ecb2656 100644 --- a/roop/ui.py +++ b/roop/ui.py @@ -199,11 +199,11 @@ def init_preview() -> None: def update_preview(frame_number: int = 0) -> None: if roop.globals.source_path and roop.globals.target_path: - video_frame = None + video_frame = get_video_frame(roop.globals.target_path, frame_number) for frame_processor in get_frame_processors_modules(roop.globals.frame_processors): video_frame = frame_processor.process_frame( get_one_face(cv2.imread(roop.globals.source_path)), - get_video_frame(roop.globals.target_path, frame_number) + video_frame ) image = Image.fromarray(cv2.cvtColor(video_frame, cv2.COLOR_BGR2RGB)) image = ImageOps.contain(image, (PREVIEW_MAX_WIDTH, PREVIEW_MAX_HEIGHT), Image.LANCZOS)