From 2e6d3dab84e3fe01122fdcf3b3791682a68ebe56 Mon Sep 17 00:00:00 2001 From: Nicolas Mowen Date: Mon, 8 Sep 2025 15:15:16 -0600 Subject: [PATCH] Always handle RKNN as NHWC in Frigate+ model loading --- frigate/detectors/detector_config.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/frigate/detectors/detector_config.py b/frigate/detectors/detector_config.py index d7883523d..7ee04bde5 100644 --- a/frigate/detectors/detector_config.py +++ b/frigate/detectors/detector_config.py @@ -161,6 +161,10 @@ class ModelConfig(BaseModel): if model_info.get("inputDataType"): self.input_dtype = model_info["inputDataType"] + # RKNN always uses NHWC + if detector == "rknn": + self.input_tensor = InputTensorEnum.nhwc + # generate list of attribute labels self.attributes_map = { **model_info.get("attributes", DEFAULT_ATTRIBUTE_LABEL_MAP),