yolov5 onnx转rknn

[复制链接] · 发表于 2023-2-27 15:27:53

用netron打开，yolov5的onnx输入是nchw，转成rknn输入也显示nchw，为什么在C++数据结构中的input_attrs[0].fmt等于RKNN_TENSOR_NHWC

只看该作者 · 发表于 2023-2-27 16:28:59

因为你的开发板和PC上存储图像的格式就是NHWC，如果不给你们转好，你要如何喂数据?

只看该作者 · 发表于 2023-2-27 17:13:34

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

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

只看该作者 · 发表于 2023-2-27 17:22:01

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

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