mirror of
https://github.com/facefusion/facefusion.git
synced 2026-08-07 09:38:39 +02:00
rename face landmarker to face aligner (#1215)
This commit is contained in:
@@ -1,7 +1,7 @@
|
||||
import numpy
|
||||
import pytest
|
||||
|
||||
from facefusion import face_classifier, face_detector, face_landmarker, face_recognizer, state_manager
|
||||
from facefusion import face_aligner, face_classifier, face_detector, face_recognizer, state_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
|
||||
@@ -26,13 +26,13 @@ def before_all() -> None:
|
||||
state_manager.init_item('face_detector_size', '640x640')
|
||||
state_manager.init_item('face_detector_margin', (0, 0, 0, 0))
|
||||
state_manager.init_item('face_detector_score', 0.5)
|
||||
state_manager.init_item('face_landmarker_model', 'many')
|
||||
state_manager.init_item('face_landmarker_score', 0.5)
|
||||
state_manager.init_item('face_aligner_model', 'many')
|
||||
state_manager.init_item('face_aligner_score', 0.5)
|
||||
state_manager.init_item('face_tracker_score', 0.3)
|
||||
|
||||
face_classifier.pre_check()
|
||||
face_detector.pre_check()
|
||||
face_landmarker.pre_check()
|
||||
face_aligner.pre_check()
|
||||
face_recognizer.pre_check()
|
||||
|
||||
|
||||
@@ -40,7 +40,7 @@ def before_all() -> None:
|
||||
def before_each() -> None:
|
||||
face_classifier.clear_inference_pool()
|
||||
face_detector.clear_inference_pool()
|
||||
face_landmarker.clear_inference_pool()
|
||||
face_aligner.clear_inference_pool()
|
||||
face_recognizer.clear_inference_pool()
|
||||
clear_faces()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user