From 0ef5de3095c510e23c2305f1d32fe0ffee103d7b Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Tue, 25 Mar 2025 11:28:14 -0600 Subject: [PATCH] Cleanup --- frigate/data_processing/common/face/model.py | 2 +- frigate/data_processing/real_time/face.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/frigate/data_processing/common/face/model.py b/frigate/data_processing/common/face/model.py index dc363a11c3..026b2f4e36 100644 --- a/frigate/data_processing/common/face/model.py +++ b/frigate/data_processing/common/face/model.py @@ -197,7 +197,7 @@ class FaceNetRecognizer(FaceRecognizer): self.face_embedder: FaceNetEmbedding = FaceNetEmbedding() def clear(self) -> None: - self.mean_embs = None + self.mean_embs = {} def build(self): if not self.landmark_detector: diff --git a/frigate/data_processing/real_time/face.py b/frigate/data_processing/real_time/face.py index c9f31a76df..348309f22a 100644 --- a/frigate/data_processing/real_time/face.py +++ b/frigate/data_processing/real_time/face.py @@ -36,8 +36,8 @@ MAX_DETECTION_HEIGHT = 1080 MIN_MATCHING_FACES = 2 -def weighted_average_by_area(results_list): - if not results_list: +def weighted_average_by_area(results_list: list[tuple[str, float, int]]): + if len(results_list) < 3: return "unknown", 0.0 score_count = {}