Toybrick

rknn_api中输入输出tensor数量的疑问

sunxing

注册会员

积分
60
楼主
发表于 2019-10-25 19:29:32    查看: 5159|回复: 2 | [复制链接]    打印 | 只看该作者
大佬们:
我看文档上说query时候能够查看输入和输出的tensor的数量,这个数量到底是什么意思?是批大小吗?感觉好像也不是,有大佬能给我详细科普下是什么意思吗?和NCHW有什么关系?
努力吧,少年。。。
回复

使用道具 举报

sunxing

注册会员

积分
60
沙发
 楼主| 发表于 2019-10-26 14:59:44 | 只看该作者
3) rknn 模型的 input/output 属性可能已经和原始模型(pb 或 caffe) 不同, 因此需要通过
rknn_query 这个 api 获取 input 和 output 的属性,如下:
rknn_input_output_num io_num;
ret = rknn_query(ctx, RKNN_QUERY_IN_OUT_NUM, &io_num, sizeof(io_num));
if(ret < 0) {
printf("rknn_query fail! ret=%d\n",ret);
goto Error;
}
上 述 接 口 用 于 获 取 input/output 的 个 数 , 个 数 会 存 储 在 io_num.n_input 和
io_num.n_output 里。

这里的input/output的个数到底是什么意思?求助大佬
努力吧,少年。。。
回复

使用道具 举报

sunxing

注册会员

积分
60
板凳
 楼主| 发表于 2019-10-29 14:35:12 | 只看该作者
hisping 发表于 2019-10-28 08:52
就是模型的输入输出个数啊,比如yolov3-tiny输入是1张图片输出是2个大数组

首先谢谢这位兄弟给我解答!
顺便问一下,我随便加载了一个转化后的模型,信息如下,输入nhwc被弄成了8个dim,前面的4个都是0,输出有4个也有两个0,这些0是怎么回事?
===========================================
input_tensors count:1
index=0,name=[]
ndims=8, format=1,type=3
0, 0, 0, 0, 3, 256, 256, 1,
===========================================
output_tensors count:1
index=0,name=[]
ndims=4, format=0,type=3
0, 0, 2, 65536,
===========================================
努力吧,少年。。。
回复

使用道具 举报

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

本版积分规则

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


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