From d459ae981e608a0d2d329c1ef292bc22bef453d2 Mon Sep 17 00:00:00 2001 From: henryruhs Date: Sat, 22 Feb 2025 23:34:46 +0100 Subject: [PATCH] Rename to detect_face_landmark() --- facefusion/face_analyser.py | 4 ++-- facefusion/face_landmarker.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/facefusion/face_analyser.py b/facefusion/face_analyser.py index a80384b..673ecbe 100644 --- a/facefusion/face_analyser.py +++ b/facefusion/face_analyser.py @@ -7,7 +7,7 @@ from facefusion.common_helper import get_first from facefusion.face_classifier import classify_face from facefusion.face_detector import detect_faces, detect_rotated_faces from facefusion.face_helper import apply_nms, convert_to_face_landmark_5, estimate_face_angle, get_nms_threshold -from facefusion.face_landmarker import detect_face_landmarks, estimate_face_landmark_68_5 +from facefusion.face_landmarker import detect_face_landmark, estimate_face_landmark_68_5 from facefusion.face_recognizer import calc_embedding from facefusion.face_store import get_static_faces, set_static_faces from facefusion.types import BoundingBox, Face, FaceLandmark5, FaceLandmarkSet, FaceScoreSet, Score, VisionFrame @@ -29,7 +29,7 @@ def create_faces(vision_frame : VisionFrame, bounding_boxes : List[BoundingBox], face_angle = estimate_face_angle(face_landmark_68_5) if state_manager.get_item('face_landmarker_score') > 0: - face_landmark_68, face_landmark_score_68 = detect_face_landmarks(vision_frame, bounding_box, face_angle) + face_landmark_68, face_landmark_score_68 = detect_face_landmark(vision_frame, bounding_box, face_angle) if face_landmark_score_68 > state_manager.get_item('face_landmarker_score'): face_landmark_5_68 = convert_to_face_landmark_5(face_landmark_68) diff --git a/facefusion/face_landmarker.py b/facefusion/face_landmarker.py index 45e9093..cab9627 100644 --- a/facefusion/face_landmarker.py +++ b/facefusion/face_landmarker.py @@ -115,7 +115,7 @@ def pre_check() -> bool: return conditional_download_hashes(model_hash_set) and conditional_download_sources(model_source_set) -def detect_face_landmarks(vision_frame : VisionFrame, bounding_box : BoundingBox, face_angle : Angle) -> Tuple[FaceLandmark68, Score]: +def detect_face_landmark(vision_frame : VisionFrame, bounding_box : BoundingBox, face_angle : Angle) -> Tuple[FaceLandmark68, Score]: face_landmark_2dfan4 = None face_landmark_peppa_wutz = None face_landmark_score_2dfan4 = 0.0