diff --git a/frigate/data_processing/common/face/model.py b/frigate/data_processing/common/face/model.py index ecdbfd8a5..ffe09eb42 100644 --- a/frigate/data_processing/common/face/model.py +++ b/frigate/data_processing/common/face/model.py @@ -267,7 +267,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: diff --git a/frigate/embeddings/onnx/face_embedding.py b/frigate/embeddings/onnx/face_embedding.py index c5d275186..a5272da0f 100644 --- a/frigate/embeddings/onnx/face_embedding.py +++ b/frigate/embeddings/onnx/face_embedding.py @@ -10,7 +10,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 @@ -110,7 +110,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", @@ -148,7 +148,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):