nik/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-08-31 07:27:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

Rename

Browse Source
This commit is contained in:
Nicolas Mowen 2025-03-25 14:28:42 -06:00
parent b359c4b86b
commit 3b77e02a40
2 changed files with 3 additions and 3 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

@ -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] = {}

4
frigate/data_processing/real_time/face.py
View File

@ -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()