From 2e51ae2eb7a4b859543602a11e137b3692fc3f3e Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Thu, 12 Dec 2024 12:53:48 -0600 Subject: [PATCH] Reimplement downloading of face detection model (#15472) --- frigate/embeddings/embeddings.py | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/frigate/embeddings/embeddings.py b/frigate/embeddings/embeddings.py index 63597e49e..9e8b73ac9 100644 --- a/frigate/embeddings/embeddings.py +++ b/frigate/embeddings/embeddings.py @@ -123,6 +123,17 @@ class Embeddings: device="GPU" if config.semantic_search.model_size == "large" else "CPU", ) + if self.config.face_recognition.enabled: + self.face_embedding = GenericONNXEmbedding( + model_name="facedet", + model_file="facedet.onnx", + download_urls={ + "facedet.onnx": "https://github.com/opencv/opencv_zoo/raw/refs/heads/main/models/face_detection_yunet/face_detection_yunet_2023mar_int8.onnx", + }, + model_type=ModelTypeEnum.face, + requestor=self.requestor, + ) + self.lpr_detection_model = None self.lpr_classification_model = None self.lpr_recognition_model = None