Toybrick

标题: onnx 模型 转 rknn 后，输入维度反了 [打印本页]
作者: ryman1981    时间: 2023-2-28 16:20
标题: onnx 模型 转 rknn 后，输入维度反了
本帖最后由 ryman1981 于 2023-2-28 16:22 编辑

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')
复制代码请问这是什么原因造成的？
作者: jefferyzhang    时间: 2023-2-28 16:42
应该是  [1,398,224,3] NHWC，你再确认下
作者: ryman1981    时间: 2023-3-1 11:06
谢谢回复！ 但是我将 input_size_list 改成 1,398,224,3 后，rknn导出的模型输入还是398,224,3,1



欢迎光临 Toybrick (https://t.rock-chips.com/) Powered by Discuz! X3.3