CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-07-28 21:08:54 +02:00
Code Issues Packages Projects Releases Wiki Activity

probe video metadata via ffprobe in vision (#1184)

Browse Source 
* probe video metadata via ffprobe in vision

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01Tbcd6VWCiU4BQP1gywPr2a

* probe video metadata via ffprobe in vision

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>
Claude-Session: https://claude.ai/code/session_01Tbcd6VWCiU4BQP1gywPr2a

---------

Co-authored-by: Claude Opus 4.8 <noreply@anthropic.com>
This commit is contained in:
Henry Ruhs
2026-07-23 12:34:32 +02:00
committed by GitHub
co-authored by Claude Opus 4.8
parent fe4aedec4d
commit a982ed9f61
6 changed files with 89 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_ffprobe.py
+11 -2
View File
@@ -19,6 +19,7 @@ def before_all() -> None:
])
subprocess.run([ 'ffmpeg', '-i', get_test_example_file('source.mp3'), '-t', '1.9', '-ar', '48000', '-ac', '2', get_test_example_file('source-48000khz-2ch.wav') ])
subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-t', '1', get_test_example_file('target-240p-1s.mkv') ])
subprocess.run([ 'ffmpeg', '-i', get_test_example_file('target-240p.mp4'), '-t', '1', get_test_example_file('target-240p-1s.mov') ])
@@ -35,7 +36,7 @@ def test_extract_audio_metadata() -> None:
assert audio_metadata.get('sample_rate') == 48000
assert audio_metadata.get('channel_total') == 2
assert audio_metadata.get('frame_total') == 91200
assert audio_metadata.get('bit_rate') == 1536000
assert audio_metadata.get('bit_rate') == 1536328
def test_extract_video_metadata() -> None:
@@ -44,7 +45,15 @@ def test_extract_video_metadata() -> None:
assert video_metadata.get('fps') == 25.0
assert video_metadata.get('duration') == 10.8
assert video_metadata.get('resolution') == (426, 226)
assert video_metadata.get('bit_rate') == 138754
assert video_metadata.get('bit_rate') == 141981
assert video_metadata.get('color_transfer') == 'smpte170m'
video_metadata = extract_video_metadata(get_test_example_file('target-240p-1s.mkv'))
assert video_metadata.get('fps') == 25.0
assert video_metadata.get('duration') == 1.0
assert video_metadata.get('frame_total') == 25
assert video_metadata.get('resolution') == (426, 226)
video_metadata = extract_video_metadata(get_test_example_file('target-240p-1s.mov'))
tests/test_ffprobe_builder.py
+12 -6
View File
@@ -1,7 +1,7 @@
from shutil import which
from facefusion import ffprobe_builder
from facefusion.ffprobe_builder import chain, format_to_key_value, run, set_input, show_entries
from facefusion.ffprobe_builder import chain, format_to_key_value, run, select_stream, set_input, show_stream_entries
def test_run() -> None:
@@ -10,15 +10,21 @@ def test_run() -> None:
def test_chain() -> None:
assert chain(
ffprobe_builder.show_entries([ 'sample_rate' ]),
ffprobe_builder.select_stream('a:0'),
ffprobe_builder.show_stream_entries([ 'sample_rate' ]),
ffprobe_builder.format_to_key_value(),
ffprobe_builder.set_input('audio.mp3')
) == [ '-show_entries', 'stream=sample_rate', '-of', 'default=noprint_wrappers=1', '-i', 'audio.mp3' ]
) == [ '-select_streams', 'a:0', '-show_entries', 'stream=sample_rate', '-of', 'default=noprint_wrappers=1', '-i', 'audio.mp3' ]
def test_show_entries() -> None:
assert show_entries([ 'duration' ]) == [ '-show_entries', 'stream=duration' ]
assert show_entries([ 'duration', 'sample_rate']) == [ '-show_entries', 'stream=duration,sample_rate' ]
def test_select_stream() -> None:
assert select_stream('a:0') == [ '-select_streams', 'a:0' ]
assert select_stream('v:0') == [ '-select_streams', 'v:0' ]
def test_show_stream_entries() -> None:
assert show_stream_entries([ 'duration' ]) == [ '-show_entries', 'stream=duration' ]
assert show_stream_entries([ 'duration', 'sample_rate' ]) == [ '-show_entries', 'stream=duration,sample_rate' ]
def test_format_to_key_value() -> None: