Toybrick

Zero-Copy API处理动态Shape输入问题

萌虎龟来

新手上路

积分
31
发表于 2024-2-23 16:46:22    查看: 909|回复: 0 | [复制链接]    打印 | 显示全部楼层
我的模型为可变Shape输入,动态输入信息查询如下:
index=0, name=input, shape_number=2, range=[[1, 255, 255, 3], [1, 127, 127, 3]], fmt=NHWC

问题:
第一步  rknn_set_input_shapes 为[1, 127, 127, 3]  模型推理成功
第二步  rknn_set_input_shapes 为[1, 255, 255, 3]  模型推理成功
第三部  rknn_set_input_shapes 为[1, 127, 127, 3]  再次设置为127时,报错如下:
E RKNN: [07:54:37.150] rknn_set_input_shapes error, input name = input supported input shapes are as follows:
E RKNN: [07:54:37.150]         shape = [1,3,255,255], layout = NHWC
E RKNN: [07:54:37.150]         shape = [1,3,127,127], layout = NHWC
E RKNN: [07:54:37.150] while get rknn_tensor_attr[0].name = input, dims = [1, 127, 127, 3], fmt = NHWC


通用API则不会出现这个问题,需要重置哪些信息来解决这个错误???

本帖子中包含更多资源

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

x
回复

使用道具 举报

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

本版积分规则

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


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