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

Asset Tools Part1 (#1217)

Browse Source 
* fix import

* introduce collect vision frame from asset, add is_vision_frame everywhere, rename to vision_buffer

* fix lint

* fix lint

* fix lint
This commit is contained in:
Henry Ruhs
2026-08-07 15:23:09 +02:00
committed by GitHub
parent a66609b388
commit 1ca152b934
9 changed files with 86 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_api_stream_video.py
+2 -3
View File
@@ -8,7 +8,6 @@ from typing import Tuple
from unittest.mock import patch
import cv2
import numpy
import pytest
from facefusion import rtc, rtc_store, state_manager
@@ -19,7 +18,7 @@ from facefusion.download import conditional_download
from facefusion.hash_helper import create_hash
from facefusion.libraries import aom as aom_module, datachannel as datachannel_module, vpx as vpx_module
from facefusion.types import Buffer, BufferPack, FrameHandler, RtcPeer, RtcPeerVideo, Time, VideoCodec
from facefusion.vision import read_video_frame
from facefusion.vision import is_vision_frame, read_video_frame
from .assert_helper import get_test_example_file, get_test_examples_directory
@@ -178,7 +177,7 @@ def test_create_and_destroy_video_decoder(video_codec : VideoCodec) -> None:
video_decoder = create_video_decoder(video_codec)
assert numpy.any(decode_video_frame(video_codec, video_decoder, encode_buffer))
assert is_vision_frame(decode_video_frame(video_codec, video_decoder, encode_buffer))
destroy_video_decoder(video_codec, video_decoder)