Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2024-11-21 19:07:46 +01:00
Code Issues Projects Releases Wiki Activity

Update facenet model (#14647)

Browse Source
This commit is contained in:
Nicolas Mowen 2024-10-28 17:35:20 -06:00
parent 57e1f30a9e
commit 86cee8e937
3 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
frigate/db/sqlitevecq.py
View File

@ -63,6 +63,6 @@ class SqliteVecQueueDatabase(SqliteQueueDatabase):
self.execute_sql(""" self.execute_sql("""
CREATE VIRTUAL TABLE IF NOT EXISTS vec_faces USING vec0( CREATE VIRTUAL TABLE IF NOT EXISTS vec_faces USING vec0(
id TEXT PRIMARY KEY, id TEXT PRIMARY KEY,
face_embedding FLOAT[128] distance_metric=cosine face_embedding FLOAT[512] distance_metric=cosine
); );
""") """)

2
frigate/embeddings/embeddings.py
View File

@ -133,7 +133,7 @@ class Embeddings:
model_name="facenet", model_name="facenet",
model_file="facenet.onnx", model_file="facenet.onnx",
download_urls={ 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", "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", model_size="large",

2
frigate/embeddings/functions/onnx.py
View File

@ -222,7 +222,7 @@ class GenericONNXEmbedding:
frame[y_center : y_center + og_h, x_center : x_center + og_w] = og frame[y_center : y_center + og_h, x_center : x_center + og_w] = og
frame = np.expand_dims(frame, axis=0) frame = np.expand_dims(frame, axis=0)
return [{"image_input": frame}] return [{"input_2": frame}]
elif self.model_type == ModelTypeEnum.alpr_detect: elif self.model_type == ModelTypeEnum.alpr_detect:
preprocessed = [] preprocessed = []
for x in raw_inputs: for x in raw_inputs: