administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-15 07:35:27 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Ensure sub label is not overwritten

Browse Source
This commit is contained in:
Nicolas Mowen 2024-10-01 07:18:09 -06:00
parent 6c971ca37a
commit 983cf53480
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
frigate/object_processing.py
View File

@ -108,7 +108,7 @@ def is_better_thumbnail(label, current_thumb, new_obj, frame_shape) -> bool:
class TrackedObject: class TrackedObject:
def __init__( def __init__(
self, camera, colormap, camera_config: CameraConfig, frame_cache, obj_data self, camera, colormap, camera_config: CameraConfig, frame_cache, obj_data: dict[str, any]
): ):
# set the score history then remove as it is not part of object state # set the score history then remove as it is not part of object state
self.score_history = obj_data["score_history"] self.score_history = obj_data["score_history"]
@ -236,6 +236,9 @@ class TrackedObject:
} }
if len(recognized_logos) > 0: if len(recognized_logos) > 0:
max_logo = max(recognized_logos, key=recognized_logos.get) max_logo = max(recognized_logos, key=recognized_logos.get)
# don't overwrite sub label if it is already set
if self.obj_data.get("sub_label") is None or self.obj_data["sub_label"][0] == max_logo:
self.obj_data["sub_label"] = (max_logo, recognized_logos[max_logo]) self.obj_data["sub_label"] = (max_logo, recognized_logos[max_logo])
# check for significant change # check for significant change