From 5a6213e2c8cea809972f7c03528fddf323597b51 Mon Sep 17 00:00:00 2001
From: Zhenpeng Zhou <zhenp3ngzhou@gmail.com>
Date: Fri, 4 Apr 2025 10:45:58 -0700
Subject: [PATCH] [Fix] Fixed the support of custom rknn models by filling the
 model type.

---
 frigate/detectors/plugins/rknn.py | 3 +++
 1 file changed, 3 insertions(+)

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