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."