Toybrick

标题: yolov5 onnx转rknn [打印本页]

作者: MMMorty 时间: 2023-2-27 15:27
标题: yolov5 onnx转rknn
用netron打开，yolov5的onnx输入是nchw，转成rknn输入也显示nchw，为什么在C++数据结构中的input_attrs[0].fmt等于RKNN_TENSOR_NHWC

作者: jefferyzhang 时间: 2023-2-27 16:28
因为你的开发板和PC上存储图像的格式就是NHWC，如果不给你们转好，你要如何喂数据?

作者: MMMorty 时间: 2023-2-27 17:13

jefferyzhang 发表于 2023-2-27 16:28
因为你的开发板和PC上存储图像的格式就是NHWC，如果不给你们转好，你要如何喂数据? ...

您的意思是说在NCHW的输入的onnx模型在转rknn的时候已经自动被转成NHWC输入了，那netron上对于rknn输入的显示是否有问题

作者: jefferyzhang 时间: 2023-2-27 17:22

MMMorty 发表于 2023-2-27 17:13
您的意思是说在NCHW的输入的onnx模型在转rknn的时候已经自动被转成NHWC输入了，那netron上对于rknn输入的 ...

显示没问题，librknn内部给你转了，你非要传NCHW是可以自己设置input attr传进去的

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