|
rk1808运行yolo4转换模型死机。
用rknn toolkit 1.6转换的yolo4模型。在虚拟环境下可以正常运行。
但如果指定目标机为1808,初始化时就死机了。请版主看看可能是什么问题。
代码:
rknn = RKNN()
RKNN_MODEL_PATH = './yolov4_tiny.rknn'
print('Loading RKNN model')
ret = rknn.load_rknn(RKNN_MODEL_PATH)
if ret != 0:
print('load rknn model failed.')
exit(ret)
print('done')
print('--> init runtime')
# ret = rknn.init_runtime()
ret = rknn.init_runtime(target = 'rk1808') //这里如果参数为空,不指定1808,可以正常运行得到结果
print(ret)
if ret != 0:
print('init runtime failed.')
exit(ret)
print('done')
运行结果:
Loading RKNN model
done
--> init runtime
I NPUTransfer: Starting NPU Transfer Client, Transfer version 2.1.0 (b5861e7@2020-11-23T11:50:36)
到这里就死机了。
|
|