fix chrash after restart (#8661)

This commit is contained in:
Marc Altmann 2023-11-18 22:36:24 +01:00 committed by GitHub
parent 678f1201c6
commit 977eef9138
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -62,10 +62,11 @@ class Rknn(DetectionApi):
) )
) )
if "rk356" in soc: if not os.path.isfile("/usr/lib/librknnrt.so"):
os.rename("/usr/lib/librknnrt_rk356x.so", "/usr/lib/librknnrt.so") if "rk356" in soc:
elif "rk3588" in soc: os.rename("/usr/lib/librknnrt_rk356x.so", "/usr/lib/librknnrt.so")
os.rename("/usr/lib/librknnrt_rk3588.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.model_path = config.model.path or "default-yolov8n"
self.core_mask = config.core_mask self.core_mask = config.core_mask