diff --git a/docs/docs/configuration/object_detectors.md b/docs/docs/configuration/object_detectors.md index 0a2cf7833..d48ae1de5 100644 --- a/docs/docs/configuration/object_detectors.md +++ b/docs/docs/configuration/object_detectors.md @@ -894,7 +894,6 @@ model: # required width: 320 height: 320 input_tensor: nhwc - input_dtype: float labelmap_path: /labelmap/coco-80.txt ``` diff --git a/frigate/detectors/plugins/rknn.py b/frigate/detectors/plugins/rknn.py index 1fdd8b9c0..828507c54 100644 --- a/frigate/detectors/plugins/rknn.py +++ b/frigate/detectors/plugins/rknn.py @@ -119,7 +119,7 @@ class Rknn(DetectionApi): model_props["model_type"] = model_type if model_matched: - model_props["filename"] = model_path + f"-{soc}-v2.3.2-1.rknn" + model_props["filename"] = model_path + f"-{soc}-v2.3.2-2.rknn" model_props["path"] = model_cache_dir + model_props["filename"] @@ -140,26 +140,10 @@ class Rknn(DetectionApi): os.mkdir(model_cache_dir) urllib.request.urlretrieve( - f"https://github.com/MarcA711/rknn-models/releases/download/v2.3.2/{filename}", + f"https://github.com/MarcA711/rknn-models/releases/download/v2.3.2-2/{filename}", model_cache_dir + filename, ) - def check_config(self, config): - if (config.model.width != 320) or (config.model.height != 320): - raise Exception( - "Make sure to set the model width and height to 320 in your config." - ) - - if config.model.input_pixel_format != "bgr": - raise Exception( - 'Make sure to set the model input_pixel_format to "bgr" in your config.' - ) - - if config.model.input_tensor != "nhwc": - raise Exception( - 'Make sure to set the model input_tensor to "nhwc" in your config.' - ) - def post_process_yolonas(self, output: list[np.ndarray]): """ @param output: output of inference