Toybrick

多输入ONNX转换RK3588问题

HurryUp

新手上路

积分
29
楼主
发表于 2024-4-10 14:23:18    查看: 1711|回复: 0 | [复制链接]    打印 | 只看该作者
目前训练了一个多模态的分类模型,有两个非图片的多维度输入,分别是float32[1,32,256]float32[1,3,16,114,114]。为了能够进行转化,将这两个维度重新导出为float32[1,1,32,256]和float32[1,48,114,114]的4维数据。设置为rknn.config(mean_values=[[123.675]], std_values=[[58.395]], target_platform=platform, optimization_level=0);ret = rknn.load_onnx(model=MODEL_PATH, inputs=['physical','video'], input_size_list=[[1,1,32,256], [1,48,114,114]], outputs=['result']);想请问针对这个模型这样设置是正确的吗?另外使用此设置在进行模型转化时报错

I     expand_to_4d_conv: remove node = [], add node = ['Conv_41_0_expand0', 'Conv_41_0_expand1']E build: Catch exception when building RKNN model!E build: Traceback (most recent call last):E build:   File "rknn/api/rknn_base.py", line 1546, in rknn.api.rknn_base.RKNNBase.buildE build:   File "rknn/api/graph_optimizer.py", line 1276, in rknn.api.graph_optimizer.GraphOptimizer.fuse_opsE build:   File "rknn/api/fuse_rules.py", line 5747, in rknn.api.fuse_rules._p_convert_global_avgpool_to_conv2E build: ValueError: need more than 3 values to unpackbuild model failed
请问该怎么解决
回复

使用道具 举报

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

本版积分规则

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


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