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

[复制链接] · 发表于 2020-12-28 14:59:18

目前使用的是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的通道顺序做处理的化应该怎样设置？

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

本帖子中包含更多资源