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
@@ -2,10 +2,9 @@
import numpy
import pytest
from facefusion import face_aligner, face_classifier, face_detector, face_recognizer, ffmpeg, ffmpeg_builder, inference_manager, process_manager, state_manager
from facefusion import face_aligner, face_classifier, face_detector, face_recognizer, face_store, ffmpeg, ffmpeg_builder, inference_manager, process_manager, state_manager
from facefusion.download import conditional_download
from facefusion.face_creator import average_face_geometry, get_many_faces, get_one_face, refill_faces
from facefusion.face_store import clear_faces
from facefusion.vision import read_static_image
from .assert_helper import get_test_example_file, get_test_examples_directory
@@ -15,6 +14,8 @@ def before_all() -> None:
state_manager.init()
inference_manager.init()
face_store.init()
process_manager.start()
conditional_download(get_test_examples_directory(),
[
@@ -56,7 +57,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_get_one_face() -> None: