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

[复制链接] · 发表于 2019-10-25 19:29:32

大佬们：
我看文档上说query时候能够查看输入和输出的tensor的数量，这个数量到底是什么意思？是批大小吗？感觉好像也不是，有大佬能给我详细科普下是什么意思吗？和NCHW有什么关系？

只看该作者 · 发表于 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的个数到底是什么意思？求助大佬

只看该作者 · 发表于 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,
===========================================