8bf9170577
* Mark as NEXT * Reduce caching to avoid RAM explosion * Reduce caching to avoid RAM explosion * Update dependencies * add face-detector-pad-factor * update facefusion.ini * fix test * change pad to margin * fix order * add prepare margin * use 50% max margin * Minor fixes part2 * Minor fixes part3 * Minor fixes part4 * Minor fixes part1 * Downgrade onnxruntime as of BiRefNet broken on CPU add test update update facefusion.ini add birefnet * rename models add more models * Fix versions * Add .claude to gitignore * add normalize color add 4 channel add colors * worflows * cleanup * cleanup * cleanup * cleanup * add more models (#961) * Fix naming * changes * Fix style and mock Gradio * Fix style and mock Gradio * Fix style and mock Gradio * apply clamp * remove clamp * Add normalizer test * Introduce sanitizer for the rescue (#963) * Introduce sanitizer for the rescue * Introduce sanitizer for the rescue * Introduce sanitizer for the rescue * prepare ffmpeg for alpha support * Some cleanup * Some cleanup * Fix CI * List as TypeAlias is not allowed (#967) * List as TypeAlias is not allowed * List as TypeAlias is not allowed * List as TypeAlias is not allowed * List as TypeAlias is not allowed * Add mpeg and mxf support (#968) * Add mpeg support * Add mxf support * Adjust fix_xxx_encoder for the new formats * Extend output pattern for batch-run (#969) * Extend output pattern for batch-run * Add {target_extension} to allowed mixed files * Catch invalid output pattern keys * alpha support * cleanup * cleanup * add ProcessorOutputs type * fix preview and streamer, support alpha for background_remover * Refactor/open close processors (#972) * Introduce open/close processors * Add locales for translator * Introduce __autoload__ for translator * More cleanup * Fix import issues * Resolve the scope situation for locals * Fix installer by not using translator * Fixes after merge * Fixes after merge * Fix translator keys in ui * Use LOCALS in installer * Update and partial fix DirectML * Use latest onnxruntime * Fix performance * Fix lint issues * fix mask * fix lint * fix lint * Remove default from translator.get() * remove 'framerate=' * fix test * Rename and reorder models * Align naming * add alpha preview * fix frame-by-frame * Add alpha effect via css * preview support alpha channel * fix preview modes * Use official assets repositories * Add support for u2net_cloth * fix naming * Add more models * Add vendor, license and year direct to the models * Add vendor, license and year direct to the models * Update dependencies, Minor CSS adjustment * Ready for 3.5.0 * Fix naming * Update about messages * Fix return * Use groups to show/hide * Update preview * Conditional merge mask * Conditional merge mask * Fix import order --------- Co-authored-by: harisreedhar <h4harisreedhar.s.s@gmail.com> Co-authored-by: Harisreedhar <46858047+harisreedhar@users.noreply.github.com>
111 lines
5.2 KiB
Python
111 lines
5.2 KiB
Python
from shutil import which
|
|
|
|
from facefusion import ffmpeg_builder
|
|
from facefusion.ffmpeg_builder import chain, concat, keep_video_alpha, run, select_frame_range, set_audio_quality, set_audio_sample_size, set_stream_mode, set_video_encoder, set_video_fps, set_video_quality
|
|
|
|
|
|
def test_run() -> None:
|
|
assert run([]) == [ which('ffmpeg'), '-loglevel', 'error' ]
|
|
|
|
|
|
def test_chain() -> None:
|
|
assert chain(
|
|
ffmpeg_builder.set_input('input.mp4'),
|
|
ffmpeg_builder.set_output('output.mp4')
|
|
) == [ '-i', 'input.mp4', 'output.mp4' ]
|
|
assert chain(
|
|
ffmpeg_builder.set_video_encoder('libx264'),
|
|
ffmpeg_builder.set_video_fps(30),
|
|
ffmpeg_builder.set_audio_encoder('aac')
|
|
) == [ '-c:v', 'libx264', '-vf', 'fps=30', '-c:a', 'aac' ]
|
|
|
|
|
|
def test_concat() -> None:
|
|
assert concat(
|
|
set_video_encoder('libvpx-vp9'),
|
|
set_video_fps(30)
|
|
) == [ '-c:v', 'libvpx-vp9', '-vf', 'fps=30' ]
|
|
assert concat(
|
|
set_video_encoder('libvpx-vp9'),
|
|
set_video_fps(30),
|
|
keep_video_alpha('libvpx-vp9')
|
|
) == [ '-c:v', 'libvpx-vp9', '-vf', 'fps=30,format=yuva420p' ]
|
|
assert concat(
|
|
select_frame_range(0, 100, 30),
|
|
keep_video_alpha('libvpx-vp9')
|
|
) == [ '-vf', 'trim=start_frame=0:end_frame=100,fps=30,format=yuva420p' ]
|
|
|
|
|
|
def test_set_stream_mode() -> None:
|
|
assert set_stream_mode('udp') == [ '-f', 'mpegts' ]
|
|
assert set_stream_mode('v4l2') == [ '-f', 'v4l2' ]
|
|
|
|
|
|
def test_select_frame_range() -> None:
|
|
assert select_frame_range(0, None, 30) == [ '-vf', 'trim=start_frame=0,fps=30' ]
|
|
assert select_frame_range(None, 100, 30) == [ '-vf', 'trim=end_frame=100,fps=30' ]
|
|
assert select_frame_range(0, 100, 30) == [ '-vf', 'trim=start_frame=0:end_frame=100,fps=30' ]
|
|
assert select_frame_range(None, None, 30) == [ '-vf', 'fps=30' ]
|
|
|
|
|
|
def test_set_audio_sample_size() -> None:
|
|
assert set_audio_sample_size(16) == [ '-f', 's16le' ]
|
|
assert set_audio_sample_size(32) == [ '-f', 's32le' ]
|
|
|
|
|
|
def test_set_audio_quality() -> None:
|
|
assert set_audio_quality('aac', 0) == [ '-q:a', '0.1' ]
|
|
assert set_audio_quality('aac', 50) == [ '-q:a', '1.0' ]
|
|
assert set_audio_quality('aac', 100) == [ '-q:a', '2.0' ]
|
|
assert set_audio_quality('libmp3lame', 0) == [ '-q:a', '9' ]
|
|
assert set_audio_quality('libmp3lame', 50) == [ '-q:a', '4' ]
|
|
assert set_audio_quality('libmp3lame', 100) == [ '-q:a', '0' ]
|
|
assert set_audio_quality('libopus', 0) == [ '-b:a', '64k' ]
|
|
assert set_audio_quality('libopus', 50) == [ '-b:a', '160k' ]
|
|
assert set_audio_quality('libopus', 100) == [ '-b:a', '256k' ]
|
|
assert set_audio_quality('libvorbis', 0) == [ '-q:a', '-1.0' ]
|
|
assert set_audio_quality('libvorbis', 50) == [ '-q:a', '4.5' ]
|
|
assert set_audio_quality('libvorbis', 100) == [ '-q:a', '10.0' ]
|
|
assert set_audio_quality('flac', 0) == []
|
|
assert set_audio_quality('flac', 50) == []
|
|
assert set_audio_quality('flac', 100) == []
|
|
|
|
|
|
def test_set_video_quality() -> None:
|
|
assert set_video_quality('libx264', 0) == [ '-crf', '51' ]
|
|
assert set_video_quality('libx264', 50) == [ '-crf', '26' ]
|
|
assert set_video_quality('libx264', 100) == [ '-crf', '0' ]
|
|
assert set_video_quality('libx264rgb', 0) == [ '-crf', '51' ]
|
|
assert set_video_quality('libx264rgb', 50) == [ '-crf', '26' ]
|
|
assert set_video_quality('libx264rgb', 100) == [ '-crf', '0' ]
|
|
assert set_video_quality('libx265', 0) == [ '-crf', '51' ]
|
|
assert set_video_quality('libx265', 50) == [ '-crf', '26' ]
|
|
assert set_video_quality('libx265', 100) == [ '-crf', '0' ]
|
|
assert set_video_quality('libvpx-vp9', 0) == [ '-crf', '63' ]
|
|
assert set_video_quality('libvpx-vp9', 50) == [ '-crf', '32' ]
|
|
assert set_video_quality('libvpx-vp9', 100) == [ '-crf', '0' ]
|
|
assert set_video_quality('h264_nvenc', 0) == [ '-cq' , '51' ]
|
|
assert set_video_quality('h264_nvenc', 50) == [ '-cq' , '26' ]
|
|
assert set_video_quality('h264_nvenc', 100) == [ '-cq' , '0' ]
|
|
assert set_video_quality('hevc_nvenc', 0) == [ '-cq' , '51' ]
|
|
assert set_video_quality('hevc_nvenc', 50) == [ '-cq' , '26' ]
|
|
assert set_video_quality('hevc_nvenc', 100) == [ '-cq' , '0' ]
|
|
assert set_video_quality('h264_amf', 0) == [ '-qp_i', '51', '-qp_p', '51', '-qp_b', '51' ]
|
|
assert set_video_quality('h264_amf', 50) == [ '-qp_i', '26', '-qp_p', '26', '-qp_b', '26' ]
|
|
assert set_video_quality('h264_amf', 100) == [ '-qp_i', '0', '-qp_p', '0', '-qp_b', '0' ]
|
|
assert set_video_quality('hevc_amf', 0) == [ '-qp_i', '51', '-qp_p', '51', '-qp_b', '51' ]
|
|
assert set_video_quality('hevc_amf', 50) == [ '-qp_i', '26', '-qp_p', '26', '-qp_b', '26' ]
|
|
assert set_video_quality('hevc_amf', 100) == [ '-qp_i', '0', '-qp_p', '0', '-qp_b', '0' ]
|
|
assert set_video_quality('h264_qsv', 0) == [ '-qp', '51' ]
|
|
assert set_video_quality('h264_qsv', 50) == [ '-qp', '26' ]
|
|
assert set_video_quality('h264_qsv', 100) == [ '-qp', '0' ]
|
|
assert set_video_quality('hevc_qsv', 0) == [ '-qp', '51' ]
|
|
assert set_video_quality('hevc_qsv', 50) == [ '-qp', '26' ]
|
|
assert set_video_quality('hevc_qsv', 100) == [ '-qp', '0' ]
|
|
assert set_video_quality('h264_videotoolbox', 0) == [ '-b:v', '1024k' ]
|
|
assert set_video_quality('h264_videotoolbox', 50) == [ '-b:v', '25768k' ]
|
|
assert set_video_quality('h264_videotoolbox', 100) == [ '-b:v', '50512k' ]
|
|
assert set_video_quality('hevc_videotoolbox', 0) == [ '-b:v', '1024k' ]
|
|
assert set_video_quality('hevc_videotoolbox', 50) == [ '-b:v', '25768k' ]
|
|
assert set_video_quality('hevc_videotoolbox', 100) == [ '-b:v', '50512k' ]
|