From c665fd2625865e8768f56cc965b6eae4bfb9363c Mon Sep 17 00:00:00 2001 From: yoni13 Date: Sun, 12 Jan 2025 01:29:24 +0800 Subject: [PATCH] Fix Please do not set this parameter on other platforms. --- machine-learning/rknn/rknnpool.py | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/machine-learning/rknn/rknnpool.py b/machine-learning/rknn/rknnpool.py index 6787fbbf87..ea5a2403e7 100644 --- a/machine-learning/rknn/rknnpool.py +++ b/machine-learning/rknn/rknnpool.py @@ -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)