onnx模型转换为rknn模型后，input shape被改变

[复制链接] · 发表于 2023-6-25 10:15:10

我有个yolov5的onnx的模型，input shape是NCHW格式的。我转换为rknn的格式模型后，在程序里读取模型的输入的时候，显示的是NHWC的了。为什么转换成rknn，会改变input shape的格式啊？

环境：
RK3588
rknn-toolkit2

只看该作者 · 发表于 2023-6-29 11:17:06

因为计算机图片格式都是NHWC的，默认都会转为NHWC部署使用，方便软件开发者喂数据。