administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-10 21:25:24 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Fix sub label checking

Browse Source
This commit is contained in:
Nicolas Mowen 2024-03-29 11:26:01 -06:00
parent 00390e2da1
commit 327ed3e0f1
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
frigate/review/maintainer.py
View File

@ -177,8 +177,8 @@ class ReviewSegmentMaintainer(threading.Thread):
for object in active_objects:
if not object["sub_label"]:
segment.detections[object["id"]] = object["label"]
elif object["sub_label"] in ALL_ATTRIBUTE_LABELS:
segment.detections[object["id"]] = object["sub_label"]
elif object["sub_label"][0] in ALL_ATTRIBUTE_LABELS:
segment.detections[object["id"]] = object["sub_label"][0]
else:
segment.detections[object["id"]] = f'{object["label"]}-verified'
@ -233,8 +233,8 @@ class ReviewSegmentMaintainer(threading.Thread):
if not object["sub_label"]:
detections[object["id"]] = object["label"]
elif object["sub_label"] in ALL_ATTRIBUTE_LABELS:
detections[object["id"]] = object["sub_label"]
elif object["sub_label"][0] in ALL_ATTRIBUTE_LABELS:
detections[object["id"]] = object["sub_label"][0]
else:
detections[object["id"]] = f'{object["label"]}-verified'