Load model by SOC name

2025-01-13 17:08:16 +08:00
parent daf886088a
commit ebdfe1b7b6
2 changed files with 7 additions and 5 deletions
@@ -20,9 +20,11 @@ try:
                break
            else:
                is_available = False
+                soc_name = None
    is_available = os.path.exists("/sys/kernel/debug/rknpu/load")
 except (FileNotFoundError, ImportError):
    is_available = False
+    soc_name = None


 def initRKNN(rknnModel="./rknnModel/yolov5s.rknn", id=0):