convert face store (#1241)

This commit is contained in:
Henry Ruhs
2026-09-12 16:32:46 +02:00
committed by GitHub
parent c3fe04cdff
commit 1052b6f532
7 changed files with 116 additions and 19 deletions
+5 -3
View File
@@ -1,11 +1,10 @@
import numpy
import pytest
from facefusion import face_aligner, face_classifier, face_detector, face_recognizer, inference_manager, state_manager, video_manager
from facefusion import face_aligner, face_classifier, face_detector, face_recognizer, face_store, inference_manager, state_manager, video_manager
from facefusion.common_helper import get_first, get_last
from facefusion.download import conditional_download
from facefusion.face_creator import get_many_faces, get_one_face
from facefusion.face_store import clear_faces
from facefusion.face_tracker import create_face_tracks, select_face_track, track_faces
from facefusion.vision import read_static_video_frame, select_video_frames
from .assert_helper import get_test_example_file, get_test_examples_directory
@@ -16,6 +15,8 @@ def before_all() -> None:
state_manager.init()
inference_manager.init()
face_store.init()
video_manager.init()
conditional_download(get_test_examples_directory(),
@@ -47,7 +48,8 @@ def before_each() -> None:
face_detector.clear_inference_pool()
face_aligner.clear_inference_pool()
face_recognizer.clear_inference_pool()
clear_faces()
face_store.clear()
def test_track_faces() -> None: