From 1ccc4f6b774a2386e4f89ef57f3eabf0e1deceb4 Mon Sep 17 00:00:00 2001 From: yakhyo Date: Tue, 25 Nov 2025 23:28:42 +0900 Subject: [PATCH] chore: Update print --- scripts/run_face_search.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/scripts/run_face_search.py b/scripts/run_face_search.py index 648af67..e230438 100644 --- a/scripts/run_face_search.py +++ b/scripts/run_face_search.py @@ -33,7 +33,7 @@ def extract_reference_embedding(detector, recognizer, image_path: str) -> np.nda return recognizer.get_normalized_embedding(image, landmarks) -def run_video(detector, recognizer, ref_embedding: np.ndarray, threshold: float = 0.4): +def run_webcam(detector, recognizer, ref_embedding: np.ndarray, threshold: float = 0.4): cap = cv2.VideoCapture(0) # 0 = default webcam if not cap.isOpened(): raise RuntimeError("Webcam could not be opened.") @@ -82,10 +82,10 @@ def main(): detector = RetinaFace() if args.detector == "retinaface" else SCRFD() recognizer = get_recognizer(args.recognizer) - print("Extracting reference embedding...") + print(f"Loading reference: {args.image}") ref_embedding = extract_reference_embedding(detector, recognizer, args.image) - run_video(detector, recognizer, ref_embedding, args.threshold) + run_webcam(detector, recognizer, ref_embedding, args.threshold) if __name__ == "__main__":