Toybrick

使用rknn_toolkit_lite2在RK3588上部署模型出现failed to submit

xavi

注册会员

积分
93
楼主
发表于 2023-4-15 20:25:59    查看: 4530|回复: 5 | [复制链接]    打印 | 只看该作者
本帖最后由 xavi 于 2023-4-15 20:35 编辑

使用rknn_toolkit_lite2在RK3588上部署模型,运行rknn_lite.inference时出现E RKNN: failed to submit! Op id:3 , op name: conv:inputs.3_Conv_pads_conv, flags: 0x5, task start:0,task number:218,run task number:218 , run task counter: 146 int status: 0模型推理精度大幅下降。
请教一下这种问题该如何解决?感谢各位大佬不吝赐教
代码如下:
import numpy as np
import platform
from rknnlite.api import RKNNLite
# decice tree for rk356x/rk3588
DEVICE_COMPATIBLE_NODE = '/proc/device-tree/compatible'
def get_host():
    # get platform and device type
    system = platform.system()
    machine = platform.machine()
    os_machine = system + '-' + machine
    if os_machine == 'Linux-aarch64':
        try:
            with open(DEVICE_COMPATIBLE_NODE) as f:
                device_compatible_str = f.read()
                if 'rk3588' in device_compatible_str:
                    host = 'RK3588'
                else:
                    host = 'RK356x'
        except IOError:
            print('Read device node {} failed.'.format(DEVICE_COMPATIBLE_NODE))
            exit(-1)
    else:
        host = os_machine
    return host

RK3588_RKNN_MODEL = '3588_saved_model_quantization_2.rknn'

if __name__ == '__main__':
    host_name = get_host()
    rknn_model = RK3588_RKNN_MODEL
    rknn_lite = RKNNLite()
    print('load RKNN model')
    ret = rknn_lite.load_rknn(rknn_model)
    if ret != 0:
        print('Load RKNN model failed')
        exit(ret)
    print('load over')


    print('init runtime environment')
    #Debian OS
    ret = rknn_lite.init_runtime(core_mask=RKNNLite.NPU_CORE_0)
    if ret != 0:
        print('Init runtime environment failed')
        exit(ret)
    print('init over')

    data_in = np.load(r'./test_data_X_nhwc.npy')
    test_Y = np.load(r'./test_data_Y.npy')
    print('running model')
    outputs = rknn_lite.inference(inputs=[data_in])
    np.save('./rk3588_result.npy', outputs[0])
    final_result = np.load(r'./rk3588_result.npy')
    preds = final_result.argmax(axis = -1)
    # print(preds)
    # print(test_Y.argmax(axis = -1))
    acc = np.mean((preds == test_Y.argmax(axis=-1)))
    print(acc)
    print('run over')










本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
回复

使用道具 举报

jefferyzhang

版主

积分
13572
沙发
发表于 2023-4-17 12:09:42 | 只看该作者
你截图的日志里已经很清楚的解释了原因
回复

使用道具 举报

xavi

注册会员

积分
93
板凳
 楼主| 发表于 2023-4-18 12:14:28 | 只看该作者
本帖最后由 xavi 于 2023-4-18 14:12 编辑
jefferyzhang 发表于 2023-4-17 12:09
你截图的日志里已经很清楚的解释了原因

感谢大佬回复,我刚上手,很多东西确实不是很熟悉,是因为1.4.0not match with rknn runtime version: 1.3.0吗?我用1.3版本的rknn试过,问题还是一样的,只不过是没有1.4.0not match with rknn runtime version: 1.3.0那段警告
这个模型我在PC上RKNN的仿真里面是跑通了的,没有failed to submit,但是在3588的板子上面用rknn-toolkit-lite跑就会出问题

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
回复

使用道具 举报

jefferyzhang

版主

积分
13572
地板
发表于 2023-4-18 14:54:04 | 只看该作者
转换模型时候指定的是3588的平台么?
回复

使用道具 举报

xavi

注册会员

积分
93
5#
 楼主| 发表于 2023-4-18 15:26:36 | 只看该作者
本帖最后由 xavi 于 2023-4-18 15:33 编辑
jefferyzhang 发表于 2023-4-18 14:54
转换模型时候指定的是3588的平台么?

是的,指定的3588平台, rknn.config(target_platform = 'rk3588'), 模型转换的脚本如下

本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有帐号?立即注册

x
回复

使用道具 举报

jefferyzhang

版主

积分
13572
6#
发表于 2023-4-18 16:18:02 | 只看该作者
1. 用最新toybrick固件测试下
2. 无效的话就在rknntoolkit2的GitHub issues里提下问题
3. 去rknn的qq群里直接找他们问下这问题。
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表