Toybrick

标题: 关于RKNN模型输入BHWC顺序问题 [打印本页]

作者: Pony 时间: 2020-12-28 14:59
标题: 关于RKNN模型输入BHWC顺序问题
目前使用的是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的通道顺序做处理的化应该怎样设置？[attach]1553[/attach]

欢迎光临 Toybrick (https://t.rock-chips.com/)