administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-18 17:14:26 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Always initialize face model on startup

Browse Source
This commit is contained in:
Nicolas Mowen 2024-12-25 06:34:30 -07:00
parent 9595777433
commit 0f4cf0fa4f
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
frigate/util/model.py
View File

@ -170,6 +170,7 @@ class FaceClassificationModel:
) )
) )
self.label_map: dict[int, str] = {} self.label_map: dict[int, str] = {}
self.__build_classifier()
def __build_classifier(self) -> None: def __build_classifier(self) -> None:
labels = [] labels = []
@ -248,6 +249,7 @@ class FaceClassificationModel:
def clear_classifier(self) -> None: def clear_classifier(self) -> None:
self.classifier = None self.classifier = None
self.labeler = None self.labeler = None
self.label_map = {}
def classify_face(self, face_image: np.ndarray) -> Optional[tuple[str, float]]: def classify_face(self, face_image: np.ndarray) -> Optional[tuple[str, float]]:
if not self.label_map: if not self.label_map: