Toybrick

标题: 模型转换config设置问题 [打印本页]

作者: zengwubin 时间: 2020-5-9 12:17
标题: 模型转换config设置问题
1.设置：rknn.config(channel_mean_value='127.5 127.5 127.5 128', reorder_channel='2 1 0') inference 输入格式为CHW, 图像数据预处理时是否需要先把图片转成CHW的格式在传给inputs？？
还是直接输入图片，NPU内部会转成CHW格式推理》

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

作者: zengwubin 时间: 2020-5-9 15:53
我的是C++代码，设置了inputs_pnet[0].fmt = RKNN_TENSOR_NCHW; 我要把图像转成CHW的话该怎么转，有没例子代码参考下，谢谢。

作者: zengwubin 时间: 2020-5-9 19:11

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

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

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