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

Formatting

Browse Source
This commit is contained in:
Nicolas Mowen 2024-10-01 07:19:00 -06:00
parent 983cf53480
commit 8cb69bbacc
3 changed files with 17 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
frigate/object_processing.py
View File

@ -108,7 +108,12 @@ 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: dict[str, any] 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"]
@ -238,7 +243,10 @@ class TrackedObject:
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 # 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: 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

8
frigate/video.py
View File

@ -752,11 +752,15 @@ def process_frames(
all_objects: list[dict[str, any]] = object_tracker.tracked_objects.values() all_objects: list[dict[str, any]] = object_tracker.tracked_objects.values()
for attributes in attribute_detections.values(): for attributes in attribute_detections.values():
for attribute in attributes: for attribute in attributes:
filtered_objects = filter(lambda o: o["label"] in attribute_detections.keys(), all_objects) filtered_objects = filter(
lambda o: o["label"] in attribute_detections.keys(), all_objects
)
selected_object_id = attribute.find_best_object(filtered_objects) selected_object_id = attribute.find_best_object(filtered_objects)
if selected_object_id is not None: if selected_object_id is not None:
detections[selected_object_id]["attributes"].append(attribute.get_tracking_data()) detections[selected_object_id]["attributes"].append(
attribute.get_tracking_data()
)
# debug object tracking # debug object tracking
if False: if False: