Toybrick

标题: onnx模型转换为rknn模型后，input shape被改变 [打印本页]

作者: xxyyzz 时间: 2023-6-25 10:15
标题: onnx模型转换为rknn模型后，input shape被改变
我有个yolov5的onnx的模型，input shape是NCHW格式的。我转换为rknn的格式模型后，在程序里读取模型的输入的时候，显示的是NHWC的了。为什么转换成rknn，会改变input shape的格式啊？

环境：
RK3588
rknn-toolkit2

作者: jefferyzhang 时间: 2023-6-29 11:17
因为计算机图片格式都是NHWC的，默认都会转为NHWC部署使用，方便软件开发者喂数据。

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