Handle init

This commit is contained in:
Nicolas Mowen 2024-10-24 15:41:16 -06:00
parent 4e6fed9a64
commit ba5ca5de40

View File

@ -347,7 +347,7 @@ class ReviewSegmentMaintainer(threading.Thread):
if len(active_objects) > 0: if len(active_objects) > 0:
detections: dict[str, str] = {} detections: dict[str, str] = {}
sub_labels = set() sub_labels = dict[str, str] = {}
zones: list[str] = [] zones: list[str] = []
severity = None severity = None
@ -358,7 +358,7 @@ class ReviewSegmentMaintainer(threading.Thread):
detections[object["id"]] = object["sub_label"][0] detections[object["id"]] = object["sub_label"][0]
else: else:
detections[object["id"]] = f'{object["label"]}-verified' detections[object["id"]] = f'{object["label"]}-verified'
sub_labels.add(object["sub_label"][0]) sub_labels[object["id"]] = object["sub_label"][0]
# if object is alert label # if object is alert label
# and has entered required zones or required zones is not set # and has entered required zones or required zones is not set
@ -566,7 +566,7 @@ class ReviewSegmentMaintainer(threading.Thread):
frame_time, frame_time,
severity, severity,
{}, {},
set(), {},
[], [],
detections, detections,
) )
@ -576,7 +576,7 @@ class ReviewSegmentMaintainer(threading.Thread):
frame_time, frame_time,
SeverityEnum.alert, SeverityEnum.alert,
{manual_info["event_id"]: manual_info["label"]}, {manual_info["event_id"]: manual_info["label"]},
set(), {},
[], [],
set(), set(),
) )