fi0r1/roop
1
0
Fork 0
mirror of https://github.com/s0md3v/roop.git synced 2025-12-07 18:38:28 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix preview (#499)

Browse Source
This commit is contained in:
K1llMan 2023-06-15 20:58:54 +03:00 committed by GitHub
parent a2b26dc966
commit d3f38c7524
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
roop/ui.py
View File

@ -199,11 +199,11 @@ def init_preview() -> None:
def update_preview(frame_number: int = 0) -> None: def update_preview(frame_number: int = 0) -> None:
if roop.globals.source_path and roop.globals.target_path: 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): for frame_processor in get_frame_processors_modules(roop.globals.frame_processors):
video_frame = frame_processor.process_frame( video_frame = frame_processor.process_frame(
get_one_face(cv2.imread(roop.globals.source_path)), 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 = Image.fromarray(cv2.cvtColor(video_frame, cv2.COLOR_BGR2RGB))
image = ImageOps.contain(image, (PREVIEW_MAX_WIDTH, PREVIEW_MAX_HEIGHT), Image.LANCZOS) image = ImageOps.contain(image, (PREVIEW_MAX_WIDTH, PREVIEW_MAX_HEIGHT), Image.LANCZOS)