administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-02-03 09:45:22 +03:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Add rockchip detector

Browse Source
This commit is contained in:
Nick Mowen 2022-11-17 07:49:32 -07:00
parent 22802c9f7e
commit 9543731f1e
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
frigate/object_detection.py
View File

@ -13,6 +13,7 @@ from setproctitle import setproctitle
from frigate.config import DetectorTypeEnum, InputTensorEnum from frigate.config import DetectorTypeEnum, InputTensorEnum
from frigate.detectors.edgetpu_tfl import EdgeTpuTfl from frigate.detectors.edgetpu_tfl import EdgeTpuTfl
from frigate.detectors.cpu_tfl import CpuTfl from frigate.detectors.cpu_tfl import CpuTfl
from frigate.detectors.rockchip import RockchipDetector
from frigate.util import EventsPerSecond, SharedMemoryFrameManager, listen, load_labels from frigate.util import EventsPerSecond, SharedMemoryFrameManager, listen, load_labels
@ -57,6 +58,10 @@ class LocalObjectDetector(ObjectDetector):
self.detect_api = EdgeTpuTfl( self.detect_api = EdgeTpuTfl(
det_device=det_device, model_config=model_config 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: else:
logger.warning( logger.warning(
"CPU detectors are not recommended and should only be used for testing or for trial purposes." "CPU detectors are not recommended and should only be used for testing or for trial purposes."