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