administrator/frigate
1
0
Fork 0
mirror of https://github.com/blakeblackshear/frigate.git synced 2026-04-15 11:32:09 +03:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Always handle RKNN as NHWC in Frigate+ model loading

Browse Source
This commit is contained in:
Nicolas Mowen 2025-09-08 15:15:16 -06:00
parent f46f8a2160
commit 2e6d3dab84
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
frigate/detectors/detector_config.py
View File

@ -161,6 +161,10 @@ class ModelConfig(BaseModel):
if model_info.get("inputDataType"): if model_info.get("inputDataType"):
self.input_dtype = model_info["inputDataType"] self.input_dtype = model_info["inputDataType"]
# RKNN always uses NHWC
if detector == "rknn":
self.input_tensor = InputTensorEnum.nhwc
# generate list of attribute labels # generate list of attribute labels
self.attributes_map = { self.attributes_map = {
**model_info.get("attributes", DEFAULT_ATTRIBUTE_LABEL_MAP), **model_info.get("attributes", DEFAULT_ATTRIBUTE_LABEL_MAP),