From 07481bebde8feec2648b953618adb6c2c3d8a547 Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Tue, 18 Mar 2025 16:08:10 -0600 Subject: [PATCH] Rework save attempts option --- frigate/config/classification.py | 4 ++-- frigate/data_processing/real_time/face.py | 3 +-- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/frigate/config/classification.py b/frigate/config/classification.py index 30cd12b7c..cbe4880a1 100644 --- a/frigate/config/classification.py +++ b/frigate/config/classification.py @@ -70,8 +70,8 @@ class FaceRecognitionConfig(FrigateBaseModel): min_area: int = Field( default=500, title="Min area of face box to consider running face recognition." ) - save_attempts: bool = Field( - default=True, title="Save images of face detections for training." + save_attempts: int = Field( + default=100, ge=0, title="Number of face attempts to save in the train tab." ) blur_confidence_filter: bool = Field( default=True, title="Apply blur quality filter to face confidence." diff --git a/frigate/data_processing/real_time/face.py b/frigate/data_processing/real_time/face.py index fa0c824e2..8b89441ae 100644 --- a/frigate/data_processing/real_time/face.py +++ b/frigate/data_processing/real_time/face.py @@ -28,7 +28,6 @@ logger = logging.getLogger(__name__) MAX_DETECTION_HEIGHT = 1080 -MAX_FACE_ATTEMPTS = 100 MIN_MATCHING_FACES = 2 @@ -512,7 +511,7 @@ class FaceRealTimeProcessor(RealTimeProcessorApi): ) # delete oldest face image if maximum is reached - if len(files) > MAX_FACE_ATTEMPTS: + if len(files) > self.config.face_recognition.save_attempts: os.unlink(os.path.join(folder, files[-1])) def expire_object(self, object_id: str):