Toybrick

标题: rv1126跑facenet转化后的rknn init失败 [打印本页]

作者: ccgunner    时间: 2021-3-25 15:50
标题: rv1126跑facenet转化后的rknn init失败
我通过rknn-toolkit-1.6.0将facenet官方的pb模型转成rknn的模型,PC仿真正常,但是连接1126板子并设置
rknn.config(channel_mean_value='0 0 0 1',
            reorder_channel='0 1 2', target_platform=['rv1126'])
ret = rknn.init_runtime(target='rv1126', device_id='2020')
其中2020是我们新刷固件之后的device name,通过adb devices 查看到的

我跑PC仿真的时候:
rknn.config(channel_mean_value='0 0 0 1',
             reorder_channel='0 1 2')
ret = rknn.init_runtime()
是正常的,跑上面那种连接板子的会报错:
--> Loading rknn model
done
I NPUTransfer: Starting NPU Transfer Client, Transfer version 2.1.0 (b5861e7@2020-11-23T11:50:36)
E RKNNAPI: rknn_init,  msg_load_ack fail, ack = 1(ACK_FAIL), expect 0(ACK_SUCC)!
E RKNNAPI: ==============================================
E RKNNAPI: RKNN VERSION:
E RKNNAPI:   API: 1.6.0 (79320de build: 2020-12-29 10:56:36)
E RKNNAPI:   DRV: 1.5.2 (e67e5cb build: 2020-12-03 15:04:52)
E RKNNAPI: ==============================================
E Catch exception when init runtime!
E Traceback (most recent call last):
E   File "rknn/api/rknn_base.py", line 844, in rknn.api.rknn_base.RKNNBase.init_runtime
E   File "rknn/api/rknn_runtime.py", line 340, in rknn.api.rknn_runtime.RKNNRuntime.build_graph
E Exception: RKNN init failed. error code: RKNN_ERR_MODEL_INVALID
E Current device id is: 2020
E Devices connected:
E ['2020']
Init runtime environment failed

请问下是什么原因



作者: sunkai    时间: 2021-3-26 10:59
facenet官方的pb模型可以直接转成功么

作者: ccgunner    时间: 2021-3-30 14:39
sunkai 发表于 2021-3-26 10:59
facenet官方的pb模型可以直接转成功么

可以的,我就是转的官方的
作者: ccgunner    时间: 2021-3-30 14:40
知道问题了,原来是在pb转rk的时候config没有配置1126
作者: sunkai    时间: 2021-3-31 09:46
你好,可以分享下转换的时候代码参数设置吗




欢迎光临 Toybrick (https://t.rock-chips.com/) Powered by Discuz! X3.3