Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-09-05 17:51:36 +02:00
Code Issues Projects Releases Wiki Activity

Replace hardcoded path with constant (#19816)

Browse Source
This commit is contained in:
Grégory Marti 2025-09-01 15:26:04 +02:00 committed by GitHub
parent 9af7246b0b
commit b86e6e484f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
frigate/data_processing/common/face/model.py
View File

@ -9,7 +9,7 @@ import numpy as np
from scipy import stats from scipy import stats
from frigate.config import FrigateConfig from frigate.config import FrigateConfig
from frigate.const import MODEL_CACHE_DIR from frigate.const import FACE_DIR, MODEL_CACHE_DIR
from frigate.embeddings.onnx.face_embedding import ArcfaceEmbedding, FaceNetEmbedding from frigate.embeddings.onnx.face_embedding import ArcfaceEmbedding, FaceNetEmbedding
from frigate.log import redirect_output_to_logger from frigate.log import redirect_output_to_logger
@ -172,7 +172,7 @@ class FaceNetRecognizer(FaceRecognizer):
face_embeddings_map: dict[str, list[np.ndarray]] = {} face_embeddings_map: dict[str, list[np.ndarray]] = {}
idx = 0 idx = 0
dir = "/media/frigate/clips/faces" dir = FACE_DIR
for name in os.listdir(dir): for name in os.listdir(dir):
if name == "train": if name == "train":
continue continue
@ -282,7 +282,7 @@ class ArcFaceRecognizer(FaceRecognizer):
face_embeddings_map: dict[str, list[np.ndarray]] = {} face_embeddings_map: dict[str, list[np.ndarray]] = {}
idx = 0 idx = 0
dir = "/media/frigate/clips/faces" dir = FACE_DIR
for name in os.listdir(dir): for name in os.listdir(dir):
if name == "train": if name == "train":
continue continue