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),