From f25cc1100ac9ab15605b254754d30595320ed1a0 Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Thu, 12 Dec 2024 11:02:10 -0700 Subject: [PATCH] Reimplement downloading of face detection model --- 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