jones/facefusion
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

More edge case testing

Browse Source
This commit is contained in:
henryruhs
2025-05-23 13:09:54 +02:00
parent 68da1cf266
commit 04319fb3c6
2 changed files with 14 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
facefusion/ffmpeg.py
View File

@@ -211,7 +211,7 @@ def merge_video(target_path : str, temp_video_fps : Fps, output_video_resolution
temp_file_path = get_temp_file_path(target_path)
temp_frames_pattern = get_temp_frames_pattern(target_path, '%08d')
if output_file_format == 'mp4' and output_video_encoder == 'rawvideo':
if output_file_format in [ 'm4v', 'mkv', 'mp4' ] and output_video_encoder == 'rawvideo':
output_video_encoder = 'libx264'
if output_file_format == 'webm':

17
tests/test_ffmpeg.py
View File

@@ -22,6 +22,8 @@ def before_all() -> None:
])
subprocess.run([ 'ffmpeg', '-i', get_test_example_file('source.mp3'), get_test_example_file('source.wav') ])
subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), get_test_example_file('target-240p.avi') ])
subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), get_test_example_file('target-240p.m4v') ])
subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), get_test_example_file('target-240p.mkv') ])
subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), get_test_example_file('target-240p.mov') ])
subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), get_test_example_file('target-240p.webm') ])
subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-vf', 'fps=25', get_test_example_file('target-240p-25fps.mp4') ])
@@ -81,15 +83,16 @@ def test_extract_frames() -> None:
def test_merge_video() -> None:
merge_set =\
[
(get_test_example_file('target-240p.avi'), 'libx264'),
(get_test_example_file('target-240p.mov'), 'libx264'),
(get_test_example_file('target-240p.mp4'), 'libx264'),
(get_test_example_file('target-240p.mp4'), 'rawvideo'),
(get_test_example_file('target-240p.webm'), 'libx264'),
(get_test_example_file('target-240p.webm'), 'rawvideo')
get_test_example_file('target-240p.avi'),
get_test_example_file('target-240p.m4v'),
get_test_example_file('target-240p.mkv'),
get_test_example_file('target-240p.mp4'),
get_test_example_file('target-240p.mov'),
get_test_example_file('target-240p.webm')
]
for target_path, output_video_encoder in merge_set:
for target_path in merge_set:
for output_video_encoder in get_available_encoder_set().get('video'):
state_manager.init_item('output_video_encoder', output_video_encoder)
create_temp_directory(target_path)
extract_frames(target_path, '452x240', 25.0, 0, 1)