huangxin/ailai_image_point_diff
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

更新cls的推理修复

Browse Source
This commit is contained in:
琉璃月光
2026-01-08 17:42:25 +08:00
parent d5457e02ca
commit 2a185affc3
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
cls_main/main_cls.py
View File

@ -46,7 +46,7 @@ def init_rknn_model(model_path):
ret = rknn_lite.load_rknn(model_path) ret = rknn_lite.load_rknn(model_path)
if ret != 0: if ret != 0:
raise RuntimeError(f'Load model failed: {ret}') raise RuntimeError(f'Load model failed: {ret}')
ret = rknn_lite.init_runtime(core_mask=RKNNLite.NPU_CORE_0) ret = rknn_lite.init_runtime()
if ret != 0: if ret != 0:
raise RuntimeError(f'Init runtime failed: {ret}') raise RuntimeError(f'Init runtime failed: {ret}')
_global_rknn_instance = rknn_lite _global_rknn_instance = rknn_lite