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

Fix Please do not set this parameter on other platforms.

Browse Source
This commit is contained in:
yoni13 2025-01-12 01:29:24 +08:00
parent c72cf61ed0
commit c665fd2625
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -34,19 +34,18 @@ def initRKNN(rknnModel="./rknnModel/yolov5s.rknn", id=0):
if soc_name in coremask_supported_socs: if soc_name in coremask_supported_socs:
if id == 0: if id == 0:
core_mask = RKNNLite.NPU_CORE_0 ret = rknn_lite.init_runtime(core_mask=RKNNLite.NPU_CORE_0)
elif id == 1: elif id == 1:
core_mask = RKNNLite.NPU_CORE_1 ret = rknn_lite.init_runtime(core_mask=RKNNLite.NPU_CORE_1)
elif id == 2: elif id == 2:
core_mask = RKNNLite.NPU_CORE_2 ret = rknn_lite.init_runtime(core_mask=RKNNLite.NPU_CORE_2)
elif id == -1: elif id == -1:
core_mask = RKNNLite.NPU_CORE_0_1_2 ret = rknn_lite.init_runtime(core_mask=RKNNLite.NPU_CORE_0_1_2)
else: else:
core_mask = None ret = rknn_lite.init_runtime()
else: else:
core_mask = None ret = rknn_lite.init_runtime() # Please do not set this parameter on other platforms.
ret = rknn_lite.init_runtime(core_mask=core_mask)
if ret != 0: if ret != 0:
print("Init runtime environment failed") print("Init runtime environment failed")
exit(ret) exit(ret)