From 9543731f1e028b56c6420925f69822f74c7d797b Mon Sep 17 00:00:00 2001 From: Nick Mowen Date: Thu, 17 Nov 2022 07:49:32 -0700 Subject: [PATCH] Add rockchip detector --- frigate/object_detection.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/frigate/object_detection.py b/frigate/object_detection.py index b4f2c2c60..232f632b1 100644 --- a/frigate/object_detection.py +++ b/frigate/object_detection.py @@ -13,6 +13,7 @@ from setproctitle import setproctitle from frigate.config import DetectorTypeEnum, InputTensorEnum from frigate.detectors.edgetpu_tfl import EdgeTpuTfl from frigate.detectors.cpu_tfl import CpuTfl +from frigate.detectors.rockchip import RockchipDetector from frigate.util import EventsPerSecond, SharedMemoryFrameManager, listen, load_labels @@ -57,6 +58,10 @@ class LocalObjectDetector(ObjectDetector): self.detect_api = EdgeTpuTfl( det_device=det_device, model_config=model_config ) + elif det_type == DetectorTypeEnum.rk1808: + self.detect_api = RockchipDetector( + det_device=det_device, model_config=model_config + ) else: logger.warning( "CPU detectors are not recommended and should only be used for testing or for trial purposes."