简单 op 通过pytorch->onnx->rknn 转化后推断结果很奇怪

请确认rknn toolkit的版本，1.0.0以前版本输出是NHWC，1.0.0之后版本输出和原始模型一致。

kkkaaa 发表于 2020-4-27 17:49
我好像发现问题所在了
rknn_input = input_tensor.numpy()[0][::-1].transpose(1,2,0)
如果事先把 rknn_in ...

rknn toolkit默认inference的data_format是nhwc