fix chrash after restart

blakeblackshear · Nov 18, 2023 · a822a51 · a822a51
1 parent 4879de2
commit a822a51
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/frigate/detectors/plugins/rknn.py b/frigate/detectors/plugins/rknn.py
@@ -62,10 +62,11 @@ def __init__(self, config: RknnDetectorConfig):
                 )
             )
 
-        if "rk356" in soc:
-            os.rename("/usr/lib/librknnrt_rk356x.so", "/usr/lib/librknnrt.so")
-        elif "rk3588" in soc:
-            os.rename("/usr/lib/librknnrt_rk3588.so", "/usr/lib/librknnrt.so")
+        if not os.path.isfile("/usr/lib/librknnrt.so"):
+            if "rk356" in soc:
+                os.rename("/usr/lib/librknnrt_rk356x.so", "/usr/lib/librknnrt.so")
+            elif "rk3588" in soc:
+                os.rename("/usr/lib/librknnrt_rk3588.so", "/usr/lib/librknnrt.so")
 
         self.model_path = config.model.path or "default-yolov8n"
         self.core_mask = config.core_mask