Toybrick

标题: api 输入输出设置 [打印本页]

作者: slkjai 时间: 2019-11-13 11:29
标题: api 输入输出设置
在转换模型的时候设置是 rknn.config(channel_mean_value='0.0 0.0 0.0 1.0', reorder_channel='0 1 2')，
使用python时结果是对的，使用c api的结果是错的

在python脚本中是使用 img = img.astype(np.float32) 转成float32, 然后传入模型
在c api 中是使用 im.convertTo(inputframe, CV_32F); 转成float32,然后传入模型，传入设置是 inputs[0].pass_through = false;
inputs[0].type = RKNN_TENSOR_FLOAT32;
inputs[0].fmt = RKNN_TENSOR_NCHW;
这是什么原因？

作者: jefferyzhang 时间: 2019-11-13 11:33
NCHW ？

作者: slkjai 时间: 2019-11-13 11:41

jefferyzhang 发表于 2019-11-13 11:33
NCHW ？

这个是caffe模型，所以使用的是NCHW,我试过NHWC,结果也不行

作者: slkjai 时间: 2019-11-13 11:45
解决了，我忘记重编译了。。。

作者: jefferyzhang 时间: 2019-11-13 12:32

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