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

Save best_box and update box on end frame

Browse Source
This commit is contained in:
Nick Mowen 2022-02-21 13:56:06 -07:00
parent 1413c00da4
commit 73c24b8e25
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
frigate/object_processing.py
View File

@ -130,6 +130,7 @@ class TrackedObject:
"region": obj_data["region"], "region": obj_data["region"],
"score": obj_data["score"], "score": obj_data["score"],
} }
self.obj_data["best_box"] = obj_data["box"]
thumb_update = True thumb_update = True
# check zones # check zones
@ -173,7 +174,7 @@ class TrackedObject:
self.current_zones = current_zones self.current_zones = current_zones
return (thumb_update, significant_change) return (thumb_update, significant_change)
def to_dict(self, include_thumbnail: bool = False): def to_dict(self, include_thumbnail: bool = False, end_frame: bool = False):
snapshot_time = ( snapshot_time = (
self.thumbnail_data["frame_time"] self.thumbnail_data["frame_time"]
if not self.thumbnail_data is None if not self.thumbnail_data is None
@ -190,7 +191,7 @@ class TrackedObject:
"start_time": self.obj_data["start_time"], "start_time": self.obj_data["start_time"],
"end_time": self.obj_data.get("end_time", None), "end_time": self.obj_data.get("end_time", None),
"score": self.obj_data["score"], "score": self.obj_data["score"],
"box": self.obj_data["box"], "box": self.obj_data.get("best_box", self.obj_data["box"]) if end_frame else self.obj_data["box"],
"area": self.obj_data["area"], "area": self.obj_data["area"],
"region": self.obj_data["region"], "region": self.obj_data["region"],
"stationary": self.obj_data["motionless_count"] "stationary": self.obj_data["motionless_count"]
@ -700,7 +701,7 @@ class TrackedObjectProcessor(threading.Thread):
f"{self.topic_prefix}/events", json.dumps(message), retain=False f"{self.topic_prefix}/events", json.dumps(message), retain=False
) )
self.event_queue.put(("end", camera, obj.to_dict(include_thumbnail=True))) self.event_queue.put(("end", camera, obj.to_dict(include_thumbnail=True, final_frame=True)))
def snapshot(camera, obj: TrackedObject, current_frame_time): def snapshot(camera, obj: TrackedObject, current_frame_time):
mqtt_config = self.config.cameras[camera].mqtt mqtt_config = self.config.cameras[camera].mqtt