From 6cecb07fe61f5b89c91027567703e26a96aac9ad Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Tue, 25 Mar 2025 12:13:03 -0600 Subject: [PATCH] Use scipy trimmed mean --- frigate/data_processing/common/face/model.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/frigate/data_processing/common/face/model.py b/frigate/data_processing/common/face/model.py index 026b2f4e36..54d95ac671 100644 --- a/frigate/data_processing/common/face/model.py +++ b/frigate/data_processing/common/face/model.py @@ -4,6 +4,7 @@ from abc import ABC, abstractmethod import cv2 import numpy as np +from scipy import stats from frigate.config import FrigateConfig from frigate.const import MODEL_CACHE_DIR @@ -234,7 +235,7 @@ class FaceNetRecognizer(FaceRecognizer): return for name, embs in face_embeddings_map.items(): - self.mean_embs[name] = np.mean(embs, axis=0) + self.mean_embs[name] = stats.trim_mean(embs, 0.1) def classify(self, face_image): if not self.landmark_detector: