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

implement birdseye modes

Browse Source
This commit is contained in:
Blake Blackshear 2021-06-11 19:02:43 -05:00
parent 94c21a6363
commit 9c5b0d4138
1 changed files with 13 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
frigate/output.py
View File

@ -76,6 +76,7 @@ class BroadcastThread(threading.Thread):
class BirdsEyeFrameManager: class BirdsEyeFrameManager:
def __init__(self, config, frame_manager: SharedMemoryFrameManager): def __init__(self, config, frame_manager: SharedMemoryFrameManager):
self.config = config self.config = config
self.mode = config.birdseye.mode
self.frame_manager = frame_manager self.frame_manager = frame_manager
width = config.birdseye.width width = config.birdseye.width
height = config.birdseye.height height = config.birdseye.height
@ -134,10 +135,20 @@ class BirdsEyeFrameManager:
copy_yuv_to_position(position, self.frame, self.layout_dim, frame, channel_dims) copy_yuv_to_position(position, self.frame, self.layout_dim, frame, channel_dims)
def camera_active(self, object_box_count, motion_box_count):
if self.mode == "continuous":
return True
if self.mode == "motion" and object_box_count + motion_box_count > 0:
return True
if self.mode == "objects" and object_box_count > 0:
return True
def update_frame(self): def update_frame(self):
# determine how many cameras are tracking objects within the last 30 seconds # determine how many cameras are tracking objects within the last 30 seconds
now = datetime.datetime.now().timestamp() now = datetime.datetime.now().timestamp()
# TODO: this needs to be based on the "mode" from the config
active_cameras = set( active_cameras = set(
[ [
cam cam
@ -236,7 +247,7 @@ class BirdsEyeFrameManager:
def update(self, camera, object_count, motion_count, frame_time, frame) -> bool: def update(self, camera, object_count, motion_count, frame_time, frame) -> bool:
# update the last active frame for the camera # update the last active frame for the camera
if object_count > 0: if self.camera_active(object_count, motion_count):
last_active_frame = self.cameras[camera]["last_active_frame"] last_active_frame = self.cameras[camera]["last_active_frame"]
# cleanup the old frame # cleanup the old frame
if last_active_frame != 0.0: if last_active_frame != 0.0: