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

请问下是什么原因

facenet官方的pb模型可以直接转成功么

sunkai 发表于 2021-3-26 10:59
facenet官方的pb模型可以直接转成功么

可以的，我就是转的官方的

知道问题了，原来是在pb转rk的时候config没有配置1126

你好，可以分享下转换的时候代码参数设置吗