Include sub_label in dict

This commit is contained in:
Nick Mowen 2022-09-23 07:07:11 -06:00
parent 938482a6da
commit 803e9b74d6

View File

@ -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"],