fix chrash after restart

This commit is contained in:
MarcA711 2023-11-18 18:12:13 +00:00
parent 4879de263b
commit a822a51833

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