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

add sub_label to camera activity

Browse Source
This commit is contained in:
Josh Hawkins 2024-05-01 07:30:45 -05:00
parent af71b9fc66
commit 05e147989b
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
frigate/object_processing.py
View File

@ -739,12 +739,14 @@ class CameraState:
if not obj.false_positive: if not obj.false_positive:
label = object_type label = object_type
sub_label = None
if obj.obj_data.get("sub_label"): if obj.obj_data.get("sub_label"):
if obj.obj_data.get("sub_label")[0] in ALL_ATTRIBUTE_LABELS: if obj.obj_data.get("sub_label")[0] in ALL_ATTRIBUTE_LABELS:
label = obj.obj_data["sub_label"][0] label = obj.obj_data["sub_label"][0]
else: else:
label = f"{object_type}-verified" label = f"{object_type}-verified"
sub_label = obj.obj_data["sub_label"][0]
camera_activity["objects"].append( camera_activity["objects"].append(
{ {
@ -754,6 +756,7 @@ class CameraState:
"area": obj.obj_data["area"], "area": obj.obj_data["area"],
"ratio": obj.obj_data["ratio"], "ratio": obj.obj_data["ratio"],
"score": obj.obj_data["score"], "score": obj.obj_data["score"],
"sub_label": sub_label,
} }
) )