Update facenet model (blakeblackshear#14647)

weitheng · Oct 28, 2024 · 54c2269 · 54c2269
1 parent a574d7b
commit 54c2269
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/frigate/db/sqlitevecq.py b/frigate/db/sqlitevecq.py
@@ -63,6 +63,6 @@ def create_embeddings_tables(self, face_recognition: bool) -> None:
             self.execute_sql("""
                 CREATE VIRTUAL TABLE IF NOT EXISTS vec_faces USING vec0(
                     id TEXT PRIMARY KEY,
-                    face_embedding FLOAT[128] distance_metric=cosine
+                    face_embedding FLOAT[512] distance_metric=cosine
                 );
             """)
diff --git a/frigate/embeddings/embeddings.py b/frigate/embeddings/embeddings.py
@@ -133,7 +133,7 @@ def __init__(self, config: FrigateConfig, db: SqliteVecQueueDatabase) -> None:
                 model_name="facenet",
                 model_file="facenet.onnx",
                 download_urls={
-                    "facenet.onnx": "https://github.com/NicolasSM-001/faceNet.onnx-/raw/refs/heads/main/faceNet.onnx",
+                    "facenet.onnx": "https://github.com/NickM-27/facenet-onnx/releases/download/v1.0/facenet.onnx",
                     "facedet.onnx": "https://github.com/opencv/opencv_zoo/raw/refs/heads/main/models/face_detection_yunet/face_detection_yunet_2023mar_int8.onnx",
                 },
                 model_size="large",

diff --git a/frigate/embeddings/functions/onnx.py b/frigate/embeddings/functions/onnx.py
@@ -222,7 +222,7 @@ def _preprocess_inputs(self, raw_inputs: any) -> any:
             frame[y_center : y_center + og_h, x_center : x_center + og_w] = og
 
             frame = np.expand_dims(frame, axis=0)
-            return [{"image_input": frame}]
+            return [{"input_2": frame}]
         elif self.model_type == ModelTypeEnum.alpr_detect:
             preprocessed = []
             for x in raw_inputs: