Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2024-11-21 19:07:46 +01:00
Code Issues Projects Releases Wiki Activity

Better review sub-labels (#14563)

Browse Source 
* Better review sub-labels

* Handle init
This commit is contained in:
Nicolas Mowen 2024-10-24 16:00:39 -06:00 committed by GitHub
parent f9fba94863
commit 4ff0c8a8d1
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
frigate/review/maintainer.py
View File

@ -51,7 +51,7 @@ class PendingReviewSegment:
frame_time: float, frame_time: float,
severity: SeverityEnum, severity: SeverityEnum,
detections: dict[str, str], detections: dict[str, str],
sub_labels: set[str], sub_labels: dict[str, str],
zones: list[str], zones: list[str],
audio: set[str], audio: set[str],
): ):
@ -135,7 +135,7 @@ class PendingReviewSegment:
ReviewSegment.data.name: { ReviewSegment.data.name: {
"detections": list(set(self.detections.keys())), "detections": list(set(self.detections.keys())),
"objects": list(set(self.detections.values())), "objects": list(set(self.detections.values())),
"sub_labels": list(self.sub_labels), "sub_labels": list(self.sub_labels.values()),
"zones": self.zones, "zones": self.zones,
"audio": list(self.audio), "audio": list(self.audio),
}, },
@ -261,7 +261,7 @@ class ReviewSegmentMaintainer(threading.Thread):
segment.detections[object["id"]] = object["sub_label"][0] segment.detections[object["id"]] = object["sub_label"][0]
else: else:
segment.detections[object["id"]] = f'{object["label"]}-verified' segment.detections[object["id"]] = f'{object["label"]}-verified'
segment.sub_labels.add(object["sub_label"][0]) segment.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
@ -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(),
) )