From 803e9b74d60239a7521a4c5920de75051e70d2ef Mon Sep 17 00:00:00 2001 From: Nick Mowen Date: Fri, 23 Sep 2022 07:07:11 -0600 Subject: [PATCH] Include sub_label in dict --- frigate/object_processing.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/frigate/object_processing.py b/frigate/object_processing.py index 3712d517c..8cd31cddd 100644 --- a/frigate/object_processing.py +++ b/frigate/object_processing.py @@ -80,6 +80,9 @@ class TrackedObject: # start the score history self.score_history = [self.obj_data["score"]] + # set blank sub label + self.obj_data["sub_label"] = None + def _is_false_positive(self): # once a true positive, always a true positive if not self.false_positive: @@ -180,6 +183,7 @@ class TrackedObject: "frame_time": self.obj_data["frame_time"], "snapshot_time": snapshot_time, "label": self.obj_data["label"], + "sub_label": self.obj_data["sub_label"], "top_score": self.top_score, "false_positive": self.false_positive, "start_time": self.obj_data["start_time"],