Cutlery/immich
1
0
Fork 1
mirror of https://github.com/immich-app/immich.git synced 2025-07-07 10:14:08 -04:00
Code Issues Packages Projects Releases Wiki Activity

switch to Runtime error instead of exit()

Browse Source
This commit is contained in:
yoni13 2025-01-14 18:49:57 +08:00
parent cb01a11f19
commit c21ce40d9c
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
machine-learning/rknn/rknnpool.py
View File

@ -6,6 +6,7 @@ from concurrent.futures import ThreadPoolExecutor
from queue import Queue from queue import Queue
import numpy as np import numpy as np
from numpy.typing import NDArray from numpy.typing import NDArray
from app.config import log
supported_socs = ["rk3562", "rk3566", "rk3568", "rk3576", "rk3588"] supported_socs = ["rk3562", "rk3566", "rk3568", "rk3576", "rk3588"]
coremask_supported_socs = ["rk3576","rk3588"] coremask_supported_socs = ["rk3576","rk3588"]
@ -25,16 +26,18 @@ try:
soc_name = None soc_name = None
is_available = is_available and os.path.exists("/sys/kernel/debug/rknpu/load") is_available = is_available and os.path.exists("/sys/kernel/debug/rknpu/load")
except (FileNotFoundError, ImportError): except (FileNotFoundError, ImportError):
log.debug("RKNN is not available")
is_available = False is_available = False
soc_name = None soc_name = None
def initRKNN(rknnModel="./rknnModel/yolov5s.rknn", id=0): def initRKNN(rknnModel="./rknnModel/yolov5s.rknn", id=0):
if not is_available:
raise RuntimeError("rknn is not available!")
rknn_lite = RKNNLite() rknn_lite = RKNNLite()
ret = rknn_lite.load_rknn(rknnModel) ret = rknn_lite.load_rknn(rknnModel)
if ret != 0: if ret != 0:
print("Load RKNN rknnModel failed") raise RuntimeError("Load RKNN rknnModel failed")
exit(ret)
if soc_name in coremask_supported_socs: if soc_name in coremask_supported_socs:
if id == 0: if id == 0:
@ -51,8 +54,7 @@ def initRKNN(rknnModel="./rknnModel/yolov5s.rknn", id=0):
ret = rknn_lite.init_runtime() # Please do not set this parameter on other platforms. ret = rknn_lite.init_runtime() # Please do not set this parameter on other platforms.
if ret != 0: if ret != 0:
print("Init runtime environment failed") raise RuntimeError("Init runtime environment failed")
exit(ret)
print(rknnModel, "\t\tdone") print(rknnModel, "\t\tdone")
return rknn_lite return rknn_lite