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

Extract frames with FPS output

Browse Source
This commit is contained in:
henryruhs 2023-07-18 10:12:18 +02:00
parent 663439fcee
commit 61b5090688
2 changed files with 15 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
roop/core.py
View File

@ -148,12 +148,13 @@ def start() -> None:
if predict_image(roop.globals.target_path):
destroy()
shutil.copy2(roop.globals.target_path, roop.globals.output_path)
# process frame
for frame_processor in get_frame_processors_modules(roop.globals.frame_processors):
update_status('Progressing...', frame_processor.NAME)
frame_processor.process_image(roop.globals.source_path, roop.globals.output_path, roop.globals.output_path)
frame_processor.post_process()
release_resources()
# validate
# validate image
if is_image(roop.globals.target_path):
update_status('Processing to image succeed!')
else:
@ -164,17 +165,23 @@ def start() -> None:
destroy()
update_status('Creating temp resources...')
create_temp(roop.globals.target_path)
update_status('Extracting frames...')
extract_frames(roop.globals.target_path)
# extract frames
if roop.globals.keep_fps:
fps = detect_fps(roop.globals.target_path)
update_status(f'Extracting frames with {fps} FPS...')
extract_frames(roop.globals.target_path, fps)
else:
update_status('Extracting frames with 30 FPS...')
extract_frames(roop.globals.target_path)
# process frame
temp_frame_paths = get_temp_frame_paths(roop.globals.target_path)
for frame_processor in get_frame_processors_modules(roop.globals.frame_processors):
update_status('Progressing...', frame_processor.NAME)
frame_processor.process_video(roop.globals.source_path, temp_frame_paths)
frame_processor.post_process()
release_resources()
# handles fps
# create video
if roop.globals.keep_fps:
update_status('Detecting FPS...')
fps = detect_fps(roop.globals.target_path)
update_status(f'Creating video with {fps} FPS...')
create_video(roop.globals.target_path, fps)
@ -191,8 +198,9 @@ def start() -> None:
else:
update_status('Restoring audio might cause issues as fps are not kept...')
restore_audio(roop.globals.target_path, roop.globals.output_path)
# clean and validate
# clean temp
clean_temp(roop.globals.target_path)
# validate video
if is_video(roop.globals.target_path):
update_status('Processing to video succeed!')
else:

6
roop/utilities.py
View File

@ -44,11 +44,7 @@ def detect_fps(target_path: str) -> float:
def extract_frames(target_path: str, fps: float = 30) -> None:
temp_directory_path = get_temp_directory_path(target_path)
command = ['-i', target_path, '-pix_fmt', 'rgb24']
if not roop.globals.keep_fps:
command.extend(['-vf', 'fps=' + str(fps)])
command.extend([os.path.join(temp_directory_path, '%04d.png')])
run_ffmpeg(command)
run_ffmpeg(['-i', target_path, '-pix_fmt', 'rgb24', '-vf', 'fps=' + str(fps), os.path.join(temp_directory_path, '%04d.png')])
def create_video(target_path: str, fps: float = 30) -> None: