Mirrors/blakeblackshear.frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2025-10-27 10:52:11 +01:00
Code Issues Projects Releases Wiki Activity

add detected_frames_processor

Browse Source
This commit is contained in:
Blake Blackshear 2020-11-01 10:10:43 -06:00
parent 03fe5158db
commit 47a9fc4292
1 changed files with 26 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
frigate/__main__.py
View File

@ -12,6 +12,9 @@ from frigate.edgetpu import EdgeTPUProcess
from frigate.http import create_app from frigate.http import create_app
from frigate.models import Event from frigate.models import Event
from frigate.mqtt import create_mqtt_client from frigate.mqtt import create_mqtt_client
from frigate.object_processing import TrackedObjectProcessor
from frigate.video import get_frame_shape
class FrigateApp(): class FrigateApp():
def __init__(self): def __init__(self):
self.stop_event = mp.Event() self.stop_event = mp.Event()
@ -33,8 +36,23 @@ class FrigateApp():
self.config = FRIGATE_CONFIG_SCHEMA(config) self.config = FRIGATE_CONFIG_SCHEMA(config)
for camera_config in self.config['cameras'].values():
if 'width' in camera_config and 'height' in camera_config:
frame_shape = (camera_config['height'], camera_config['width'], 3)
else:
frame_shape = get_frame_shape(camera_config['ffmpeg']['input'])
camera_config['frame_shape'] = frame_shape
# TODO: sub in FRIGATE_ENV vars # TODO: sub in FRIGATE_ENV vars
def init_queues(self):
# Queue for clip processing
self.event_queue = mp.Queue()
# Queue for cameras to push tracked objects to
self.detected_frames_queue = mp.Queue(maxsize=len(self.config['cameras'].keys())*2)
def init_database(self): def init_database(self):
self.db = SqliteExtDatabase(f"/{os.path.join(self.config['save_clips']['clips_dir'], 'frigate.db')}") self.db = SqliteExtDatabase(f"/{os.path.join(self.config['save_clips']['clips_dir'], 'frigate.db')}")
models = [Event] models = [Event]
@ -70,8 +88,10 @@ class FrigateApp():
if detector['type'] == 'edgetpu': if detector['type'] == 'edgetpu':
self.detectors[name] = EdgeTPUProcess(self.detection_queue, out_events=self.detection_out_events, tf_device=detector['device']) self.detectors[name] = EdgeTPUProcess(self.detection_queue, out_events=self.detection_out_events, tf_device=detector['device'])
def start_detection_processor(self): def start_detected_frames_processor(self):
pass self.detected_frames_processor = TrackedObjectProcessor(self.config['cameras'], self.mqtt_client, self.config['mqtt']['topic_prefix'],
self.detected_frames_queue, self.event_queue, self.stop_event)
self.detected_frames_processor.start()
def start_frame_processors(self): def start_frame_processors(self):
pass pass
@ -84,11 +104,12 @@ class FrigateApp():
def start(self): def start(self):
self.init_config() self.init_config()
self.init_queues()
self.init_database() self.init_database()
self.init_web_server() self.init_web_server()
self.init_mqtt() self.init_mqtt()
self.start_detectors() self.start_detectors()
self.start_detection_processor() self.start_detected_frames_processor()
self.start_frame_processors() self.start_frame_processors()
self.start_camera_capture_processes() self.start_camera_capture_processes()
self.start_watchdog() self.start_watchdog()
@ -98,6 +119,8 @@ class FrigateApp():
def stop(self): def stop(self):
self.stop_event.set() self.stop_event.set()
self.detected_frames_processor.join()
for detector in self.detectors.values(): for detector in self.detectors.values():
detector.stop() detector.stop()