diff --git a/face_recognition.py b/face_recognition.py index a26a4bb..c9453ec 100644 --- a/face_recognition.py +++ b/face_recognition.py @@ -24,12 +24,12 @@ FACE_DETECT_INTERVAL = 2.0 # seconds between face detection runs in process_fra class FaceRecognizer: - def __init__(self, face_model_path, embed_model_path, db_path="faces.db"): + def __init__(self, face_model_path, embed_model_path, db_path="faces.db", coral_device=2): self._lock = threading.Lock() - # Coral face detector - logger.info("Loading face detection model on Edge TPU...") - delegate = tfl.load_delegate("libedgetpu.so.1") + # Coral face detector (device index avoids conflict with YAMNet on :0 and pose on :1) + logger.info("Loading face detection model on Edge TPU (device :%d)...", coral_device) + delegate = tfl.load_delegate("libedgetpu.so.1", options={"device": f":{coral_device}"}) self._face_interp = tfl.Interpreter( model_path=str(face_model_path), experimental_delegates=[delegate],