Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-01-07 00:06:57 +01:00
Code Issues Projects Releases Wiki Activity

Use timestamp decorator helper in codebase

Browse Source
This commit is contained in:
Sebastian Englbrecht 2021-04-28 19:54:25 +02:00 committed by Blake Blackshear
parent 7bfe8e4f5b
commit 5075e4eee1
1 changed files with 18 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
frigate/object_processing.py
View File

@ -20,7 +20,7 @@ import numpy as np
from frigate.config import FrigateConfig, CameraConfig
from frigate.const import RECORD_DIR, CLIPS_DIR, CACHE_DIR
from frigate.edgetpu import load_labels
from frigate.util import SharedMemoryFrameManager, draw_box_with_label, calculate_region
from frigate.util import SharedMemoryFrameManager, draw_box_with_label, draw_timestamp, calculate_region
logger = logging.getLogger(__name__)
@ -268,23 +268,15 @@ class TrackedObject:
)
if timestamp:
time_to_show = datetime.datetime.fromtimestamp(
self.thumbnail_data["frame_time"]
).strftime("%m/%d/%Y %H:%M:%S")
size = cv2.getTextSize(
time_to_show, cv2.FONT_HERSHEY_SIMPLEX, fontScale=1, thickness=2
)
text_width = size[0][0]
desired_size = max(150, 0.33 * best_frame.shape[1])
font_scale = desired_size / text_width
cv2.putText(
best_frame,
time_to_show,
(5, best_frame.shape[0] - 7),
cv2.FONT_HERSHEY_SIMPLEX,
fontScale=font_scale,
color=(255, 255, 255),
thickness=2,
draw_timestamp(
frame_copy,
self.thumbnail_data["frame_time"],
"%m/%d/%Y %H:%M:%S",
font_effect=None,
font_scale=1.0,
font_thickness=2,
font_color=(255, 255, 255),
position="ul",
)
ret, jpg = cv2.imencode(".jpg", best_frame, [int(cv2.IMWRITE_JPEG_QUALITY), 70])
@ -407,17 +399,15 @@ class CameraState:
)
if draw_options.get("timestamp"):
time_to_show = datetime.datetime.fromtimestamp(frame_time).strftime(
"%m/%d/%Y %H:%M:%S"
)
cv2.putText(
draw_timestamp(
frame_copy,
time_to_show,
(10, 30),
cv2.FONT_HERSHEY_SIMPLEX,
fontScale=0.8,
color=(255, 255, 255),
thickness=2,
frame_time,
"%m/%d/%Y %H:%M:%S",
font_effect=None,
font_scale=1.0,
font_thickness=2,
font_color= (255, 255, 255),
position="ul",
)
return frame_copy