mirror of
https://github.com/facefusion/facefusion.git
synced 2026-09-20 22:40:42 +02:00
convert face store (#1241)
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user