Toybrick

api 输入输出设置

slkjai

注册会员

积分
60
楼主
发表于 2019-11-13 11:29:12    查看: 5675|回复: 4 | [复制链接]    打印 | 只看该作者
在转换模型的时候设置是 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

版主

积分
12940
沙发
发表于 2019-11-13 11:33:26 | 只看该作者
NCHW ?
回复

使用道具 举报

slkjai

注册会员

积分
60
板凳
 楼主| 发表于 2019-11-13 11:41:33 | 只看该作者

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

使用道具 举报

slkjai

注册会员

积分
60
地板
 楼主| 发表于 2019-11-13 11:45:29 | 只看该作者
解决了,我忘记重编译了。。。
回复

使用道具 举报

jefferyzhang

版主

积分
12940
5#
发表于 2019-11-13 12:32:47 | 只看该作者
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

产品中心 购买渠道 开源社区 Wiki教程 资料下载 关于Toybrick


快速回复 返回顶部 返回列表