From ba5b25474bcee7dd73e4c4990312868f37d5e21d Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Wed, 23 Apr 2025 08:03:45 -0600 Subject: [PATCH] Add other rockchip download models --- frigate/detectors/plugins/rknn.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/frigate/detectors/plugins/rknn.py b/frigate/detectors/plugins/rknn.py index 7794172c19..10c0742625 100644 --- a/frigate/detectors/plugins/rknn.py +++ b/frigate/detectors/plugins/rknn.py @@ -19,7 +19,11 @@ DETECTOR_KEY = "rknn" supported_socs = ["rk3562", "rk3566", "rk3568", "rk3576", "rk3588"] -supported_models = {ModelTypeEnum.yolonas: "^deci-fp16-yolonas_[sml]$"} +supported_models = { + ModelTypeEnum.yologeneric: "^frigate-fp16-yolov9-[cemst]$", + ModelTypeEnum.yolonas: "^deci-fp16-yolonas_[sml]$", + ModelTypeEnum.yolox: "^rock-(fp16|i8)-yolox_(nano|tiny)$", +} model_cache_dir = os.path.join(MODEL_CACHE_DIR, "rknn_cache/")