administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-04-27 09:07:41 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

pass face recognition config instead of whole config

Browse Source
This commit is contained in:
baudneo 2025-08-01 17:30:15 -06:00
parent 052c33ac15
commit 4bc3545835
No known key found for this signature in database
GPG Key ID: 51445F2ED08EBC7F
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -269,7 +269,7 @@ class ArcFaceRecognizer(FaceRecognizer):
def __init__(self, config: FrigateConfig):
super().__init__(config)
self.mean_embs: dict[int, np.ndarray] = {}
self.face_embedder: ArcfaceEmbedding = ArcfaceEmbedding(config)
self.face_embedder: ArcfaceEmbedding = ArcfaceEmbedding(config.face_recognition)
self.model_builder_queue: queue.Queue | None = None
def clear(self) -> None:

6
frigate/embeddings/onnx/face_embedding.py
View File

@ -11,7 +11,7 @@ from frigate.util.downloader import ModelDownloader
from .base_embedding import BaseEmbedding
from .runner import ONNXModelRunner
from ...config import FrigateConfig
from ...config import FaceRecognitionConfig
try:
from tflite_runtime.interpreter import Interpreter
@ -112,7 +112,7 @@ class FaceNetEmbedding(BaseEmbedding):
class ArcfaceEmbedding(BaseEmbedding):
def __init__(self, config: FrigateConfig):
def __init__(self, config: FaceRecognitionConfig):
super().__init__(
model_name="facedet",
model_file="arcface.onnx",
@ -150,7 +150,7 @@ class ArcfaceEmbedding(BaseEmbedding):
self.runner = ONNXModelRunner(
os.path.join(self.download_path, self.model_file),
device=self.config.face_recognition.device or "GPU",
device=self.config.device or "GPU",
)
def _preprocess_inputs(self, raw_inputs):