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

Add fps parameter to extract_frames()

Browse Source
This commit is contained in:
henryruhs 2023-07-16 15:38:48 +02:00
parent e1d7f00d48
commit 77cd226da4
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
roop/utilities.py
View File

@ -39,19 +39,19 @@ def detect_fps(target_path: str) -> float:
return numerator / denominator return numerator / denominator
except Exception: except Exception:
pass pass
return 30.0 return 30
def extract_frames(target_path: str) -> None: def extract_frames(target_path: str, fps: float = 30) -> None:
temp_directory_path = get_temp_directory_path(target_path) temp_directory_path = get_temp_directory_path(target_path)
command = ['-i', target_path, '-pix_fmt', 'rgb24'] command = ['-i', target_path, '-pix_fmt', 'rgb24']
if not roop.globals.keep_fps: if not roop.globals.keep_fps:
command.extend(['-vf', 'fps=30']) command.extend(['-vf', 'fps=' + str(fps)])
command.extend([os.path.join(temp_directory_path, '%04d.png')]) command.extend([os.path.join(temp_directory_path, '%04d.png')])
run_ffmpeg(command) run_ffmpeg(command)
def create_video(target_path: str, fps: float = 30.0) -> None: def create_video(target_path: str, fps: float = 30) -> None:
temp_output_path = get_temp_output_path(target_path) temp_output_path = get_temp_output_path(target_path)
temp_directory_path = get_temp_directory_path(target_path) temp_directory_path = get_temp_directory_path(target_path)
run_ffmpeg(['-r', str(fps), '-i', os.path.join(temp_directory_path, '%04d.png'), '-c:v', roop.globals.video_encoder, '-crf', str(roop.globals.video_quality), '-pix_fmt', 'yuv420p', '-vf', 'colorspace=bt709:iall=bt601-6-625:fast=1', '-y', temp_output_path]) run_ffmpeg(['-r', str(fps), '-i', os.path.join(temp_directory_path, '%04d.png'), '-c:v', roop.globals.video_encoder, '-crf', str(roop.globals.video_quality), '-pix_fmt', 'yuv420p', '-vf', 'colorspace=bt709:iall=bt601-6-625:fast=1', '-y', temp_output_path])