模型转换config设置问题

[复制链接] · 发表于 2020-5-9 12:17:40

1.设置：rknn.config(channel_mean_value='127.5 127.5 127.5 128', reorder_channel='2 1 0') inference 输入格式为CHW, 图像数据预处理时是否需要先把图片转成CHW的格式在传给inputs？？
还是直接输入图片，NPU内部会转成CHW格式推理》

只看该作者 · 发表于 2020-5-9 14:17:19

这个看你模型，caffe就是CHW，tf、pytorch就是HWC。NPU不会去转

只看该作者 · 发表于 2020-5-9 15:53:55

我的是C++代码，设置了inputs_pnet[0].fmt = RKNN_TENSOR_NCHW; 我要把图像转成CHW的话该怎么转，有没例子代码参考下，谢谢。

只看该作者 · 发表于 2020-5-9 19:11:05

jefferyzhang 发表于 2020-5-9 14:17
这个看你模型，caffe就是CHW，tf、pytorch就是HWC。NPU不会去转

我的是C++代码，设置了inputs_pnet[0].fmt = RKNN_TENSOR_NCHW; 我要把图像转成CHW的话该怎么转，有没例子代码参考下，谢谢。