administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-12-10 15:25:43 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Added new topic which publishes image as base64 with meta data about the event

Browse Source
This commit is contained in:
Vinod S R 2020-09-09 19:50:36 +05:30
parent 1915f18143
commit 8d3edc4050
1 changed files with 21 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
frigate/object_processing.py
View File

@ -13,6 +13,7 @@ import pyarrow.plasma as plasma
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from frigate.util import draw_box_with_label, PlasmaManager from frigate.util import draw_box_with_label, PlasmaManager
from frigate.edgetpu import load_labels from frigate.edgetpu import load_labels
import base64
PATH_TO_LABELS = '/labelmap.txt' PATH_TO_LABELS = '/labelmap.txt'
@ -262,6 +263,16 @@ class TrackedObjectProcessor(threading.Thread):
ret, jpg = cv2.imencode('.jpg', best_frame) ret, jpg = cv2.imencode('.jpg', best_frame)
if ret: if ret:
jpg_bytes = jpg.tobytes() jpg_bytes = jpg.tobytes()
jpg_as_text = base64.b64encode(jpg).decode()
payload={
"image": jpg_as_text,
"status":"ON",
"label":obj["label"],
"score": obj["score"],
"start_time" : obj["start_time"],
"id": obj["id"]
}
self.client.publish(f"{self.topic_prefix}/{camera}/{obj_name}/event", json.dumps(payload), retain=True)
self.client.publish(f"{self.topic_prefix}/{camera}/{obj_name}/snapshot", jpg_bytes, retain=True) self.client.publish(f"{self.topic_prefix}/{camera}/{obj_name}/snapshot", jpg_bytes, retain=True)
# expire any objects that are ON and no longer detected # expire any objects that are ON and no longer detected
@ -274,4 +285,14 @@ class TrackedObjectProcessor(threading.Thread):
ret, jpg = cv2.imencode('.jpg', best_frame) ret, jpg = cv2.imencode('.jpg', best_frame)
if ret: if ret:
jpg_bytes = jpg.tobytes() jpg_bytes = jpg.tobytes()
jpg_as_text = base64.b64encode(jpg).decode()
payload={
"image": jpg_as_text,
"status":"OFF",
"label":obj["label"],
"score": obj["score"],
"start_time" : obj["start_time"],
"id": obj["id"]
}
self.client.publish(f"{self.topic_prefix}/{camera}/{obj_name}/event", json.dumps(payload), retain=True)
self.client.publish(f"{self.topic_prefix}/{camera}/{obj_name}/snapshot", jpg_bytes, retain=True) self.client.publish(f"{self.topic_prefix}/{camera}/{obj_name}/snapshot", jpg_bytes, retain=True)