onnx 模型 转 rknn 后，输入维度反了

onnx模型输入维度是 [1,3, 224, 398]在我转到rknn格式后输入维度成了 [398,224,3,1] 转换过程也没有报错和警告。

1. rknn.config(reorder_channel='1 2 0', optimization_level=3, target_platform = 'rk3399pro',output_optimize=1)
   
2. print('--> Loading model')
   
3. ret = rknn.load_onnx(model=ONNX_MODEL, input_size_list=[1, 3, 224, 398])
   
4. if ret != 0:
   
5.     print('Load model failed!')
   
6.     exit(ret)
   
7. print('done')

复制代码请问这是什么原因造成的？

应该是  [1,398,224,3] NHWC，你再确认下

谢谢回复！ 但是我将 input_size_list 改成 1,398,224,3 后，rknn导出的模型输入还是398,224,3,1