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

changes for vp9 support

Browse Source
This commit is contained in:
henryruhs
2026-06-06 12:39:14 +02:00
parent 5287ce6dcb
commit f3bc5ffb4b
7 changed files with 61 additions and 68 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_api_stream_video.py
+5 -12
View File
@@ -11,7 +11,7 @@ import pytest
from facefusion import rtc, rtc_store, state_manager
from facefusion.apis.stream_video import create_video_decoder, create_video_encoder, decode_video_frame, destroy_video_decoder, destroy_video_encoder, encode_video_frame, handle_video_frame, receive_video_frames, run_video_encode_loop, update_video_encoder_bitrate
from facefusion.codecs import aom_encoder, vp9_encoder, vpx_encoder
from facefusion.codecs import aom_encoder, vpx_encoder
from facefusion.common_helper import is_linux, is_macos, is_windows
from facefusion.download import conditional_download
from facefusion.hash_helper import create_hash
@@ -165,12 +165,9 @@ def test_create_and_destroy_video_decoder(video_codec : VideoCodec) -> None:
if video_codec == 'av1':
video_encoder = aom_encoder.create((426, 226), 1000, 1, 0)
encode_buffer = aom_encoder.encode(video_encoder, input_buffer, (426, 226), 0)
if video_codec == 'vp8':
video_encoder = vpx_encoder.create((426, 226), 1000, 1, 0)
if video_codec in [ 'vp8', 'vp9' ]:
video_encoder = vpx_encoder.create(video_codec, (426, 226), 1000, 1, 0)
encode_buffer = vpx_encoder.encode(video_encoder, input_buffer, (426, 226), 0)
if video_codec == 'vp9':
video_encoder = vp9_encoder.create((426, 226), 1000, 1, 0)
encode_buffer = vp9_encoder.encode(video_encoder, input_buffer, (426, 226), 0)
video_decoder = create_video_decoder(video_codec)
@@ -189,19 +186,15 @@ def test_create_and_destroy_video_encoder(video_codec : VideoCodec) -> None:
if video_codec == 'av1':
assert aom_encoder.encode(video_encoder, input_buffer, (426, 226), 0)
if video_codec == 'vp8':
if video_codec in [ 'vp8', 'vp9' ]:
assert vpx_encoder.encode(video_encoder, input_buffer, (426, 226), 0)
if video_codec == 'vp9':
assert vp9_encoder.encode(video_encoder, input_buffer, (426, 226), 0)
destroy_video_encoder(video_codec, video_encoder)
if video_codec == 'av1':
assert aom_encoder.encode(video_encoder, input_buffer, (426, 226), 1) == bytes()
if video_codec == 'vp8':
if video_codec in [ 'vp8', 'vp9' ]:
assert vpx_encoder.encode(video_encoder, input_buffer, (426, 226), 1) == bytes()
if video_codec == 'vp9':
assert vp9_encoder.encode(video_encoder, input_buffer, (426, 226), 1) == bytes()
@pytest.mark.parametrize('video_codec', [ 'av1', 'vp8', 'vp9' ])