From 895afcdb0e3b22fccb55cb0303fc242f8ccaa6ff Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Sun, 4 May 2025 09:33:27 -0600 Subject: [PATCH] Various Fixes (#18035) * Support multi and single core rknn npus * Update docs config to be more clear --- docs/docs/configuration/object_detectors.md | 6 +++--- frigate/util/services.py | 8 +++++++- 2 files changed, 10 insertions(+), 4 deletions(-) diff --git a/docs/docs/configuration/object_detectors.md b/docs/docs/configuration/object_detectors.md index cdfe6579e..60c1e512d 100644 --- a/docs/docs/configuration/object_detectors.md +++ b/docs/docs/configuration/object_detectors.md @@ -152,7 +152,7 @@ Use this configuration for YOLO-based models. When no custom model path or URL i ```yaml detectors: - hailo8l: + hailo: type: hailo8l device: PCIe @@ -185,7 +185,7 @@ For SSD-based models, provide either a model path or URL to your compiled SSD mo ```yaml detectors: - hailo8l: + hailo: type: hailo8l device: PCIe @@ -209,7 +209,7 @@ The Hailo detector supports all YOLO models compiled for Hailo hardware that inc ```yaml detectors: - hailo8l: + hailo: type: hailo8l device: PCIe diff --git a/frigate/util/services.py b/frigate/util/services.py index 7210837ff..1c778bac4 100644 --- a/frigate/util/services.py +++ b/frigate/util/services.py @@ -408,7 +408,13 @@ def get_rockchip_npu_stats() -> dict[str, str]: try: with open("/sys/kernel/debug/rknpu/load", "r") as f: npu_output = f.read() - core_loads = re.findall(r"Core\d+:\s*(\d+)%", npu_output) + + if "Core0:" in npu_output: + # multi core NPU + core_loads = re.findall(r"Core\d+:\s*(\d+)%", npu_output) + else: + # single core NPU + core_loads = re.findall(r"NPU load:\s+(\d+)%", npu_output) except FileNotFoundError: core_loads = None