diff --git a/frigate/data_processing/common/face/model.py b/frigate/data_processing/common/face/model.py index 2af55c908c..56eda44919 100644 --- a/frigate/data_processing/common/face/model.py +++ b/frigate/data_processing/common/face/model.py @@ -198,7 +198,7 @@ class LBPHRecognizer(FaceRecognizer): return self.label_map[index], round(score, 2) -class FaceNetRecognizer(FaceRecognizer): +class ArcFaceRecognizer(FaceRecognizer): def __init__(self, config: FrigateConfig): super().__init__(config) self.mean_embs: dict[int, np.ndarray] = {} diff --git a/frigate/data_processing/real_time/face.py b/frigate/data_processing/real_time/face.py index 93de32a68d..e20dad6338 100644 --- a/frigate/data_processing/real_time/face.py +++ b/frigate/data_processing/real_time/face.py @@ -20,7 +20,7 @@ from frigate.comms.event_metadata_updater import ( from frigate.config import FrigateConfig from frigate.const import FACE_DIR, MODEL_CACHE_DIR from frigate.data_processing.common.face.model import ( - FaceNetRecognizer, + ArcFaceRecognizer, FaceRecognizer, LBPHRecognizer, ) @@ -110,7 +110,7 @@ class FaceRealTimeProcessor(RealTimeProcessorApi): if self.face_config.model_size == "small": self.recognizer = LBPHRecognizer(self.config) else: - self.recognizer = FaceNetRecognizer(self.config) + self.recognizer = ArcFaceRecognizer(self.config) self.recognizer.build()