ZF/uniface
1
0
Fork 0
mirror of https://github.com/yakhyo/uniface.git synced 2025-12-30 09:02:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

chore: Update print

Browse Source
This commit is contained in:
yakhyo
2025-11-25 23:28:42 +09:00
parent 189755a1a6
commit 1ccc4f6b77
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/run_face_search.py
View File

@@ -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__":