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

Remove limiting choices, Add error handling to start()

Browse Source
This commit is contained in:
henryruhs 2023-08-07 22:47:34 +02:00
parent 9f84898724
commit 5c1c5636c0
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
roop/core.py
View File

@ -160,12 +160,10 @@ def start() -> None:
if roop.globals.keep_fps:
fps = detect_fps(roop.globals.target_path)
update_status(f'Extracting frames with {fps} FPS...')
if not extract_frames(roop.globals.target_path, fps):
update_status('Extracting frames failed...')
extract_frames(roop.globals.target_path, fps)
else:
update_status('Extracting frames with 30 FPS...')
if not extract_frames(roop.globals.target_path):
update_status('Extracting frames failed...')
extract_frames(roop.globals.target_path)
# process frame
temp_frame_paths = get_temp_frame_paths(roop.globals.target_path)
if temp_frame_paths:
@ -174,7 +172,7 @@ def start() -> None:
frame_processor_module.process_video(roop.globals.source_path, temp_frame_paths)
frame_processor_module.post_process()
else:
update_status('Frames not found...')
update_status('Temporary frames not found...')
return
# create video
if roop.globals.keep_fps: