diff --git a/frigate/detectors/plugins/rknn.py b/frigate/detectors/plugins/rknn.py index 407c93917a..92a72fbaad 100644 --- a/frigate/detectors/plugins/rknn.py +++ b/frigate/detectors/plugins/rknn.py @@ -85,6 +85,9 @@ class Rknn(DetectionApi): if "/" in model_path: model_props["preset"] = False model_props["path"] = model_path + for model_type in ModelTypeEnum: + if model_type in model_path: + model_props["model_type"] = model_type else: model_props["preset"] = True