Toybrick

关于RKNN模型输入BHWC顺序问题

Pony

注册会员

积分
138
发表于 2020-12-28 14:59:18    查看: 6515|回复: 0 | [复制链接]    打印 | 显示全部楼层
目前使用的是RKNN-API的C接口,API版本1.3.0, DRV版本1.3.1.

请问RKNN的输入tensor维度是怎么规定的,rknn-api demo里面的设置都是NHWC,但是我的原始ONNX或者pytorch模型输入都是NCHW,期望输出也是NCHW。所以这里的网络有一个默认的通道变换吗?另外在RKNN-TOOLKIT 1.4里面是不是也有一个默认的设定呢?如果我想要模型按照NCHW的通道顺序做处理的化应该怎样设置?

本帖子中包含更多资源

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

x
回复

使用道具 举报

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

本版积分规则

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


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