Cutlery/immich
1
0
Fork 1
mirror of https://github.com/immich-app/immich.git synced 2025-06-03 13:44:16 -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 id == 0:
core_mask = RKNNLite.NPU_CORE_0
ret = rknn_lite.init_runtime(core_mask=RKNNLite.NPU_CORE_0)
elif id == 1:
core_mask = RKNNLite.NPU_CORE_1
ret = rknn_lite.init_runtime(core_mask=RKNNLite.NPU_CORE_1)
elif id == 2:
core_mask = RKNNLite.NPU_CORE_2
ret = rknn_lite.init_runtime(core_mask=RKNNLite.NPU_CORE_2)
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:
core_mask = None
ret = rknn_lite.init_runtime()
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:
print("Init runtime environment failed")
exit(ret)