From c5a83575f078d9bb25bbfecb7abbd3524b29384b Mon Sep 17 00:00:00 2001 From: Jason Hunter Date: Wed, 12 Jun 2024 15:58:13 -0400 Subject: [PATCH] only embed objects --- frigate/embeddings/maintainer.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/frigate/embeddings/maintainer.py b/frigate/embeddings/maintainer.py index a2574d886..934e16654 100644 --- a/frigate/embeddings/maintainer.py +++ b/frigate/embeddings/maintainer.py @@ -52,9 +52,9 @@ class EmbeddingMaintainer(threading.Thread): continue source_type, _, camera, data = update - camera_config = self.config.cameras[camera] - if source_type == EventTypeEnum.tracked_object: + if camera and source_type == EventTypeEnum.tracked_object: + camera_config = self.config.cameras[camera] if data["id"] not in self.tracked_events: self.tracked_events[data["id"]] = [] @@ -86,6 +86,10 @@ class EmbeddingMaintainer(threading.Thread): except DoesNotExist: continue + # Skip the event if not an object + if event.data.get("type") != "object": + continue + # Extract valid event metadata metadata = get_metadata(event) thumbnail = base64.b64decode(event.thumbnail)