Toybrick

关于C++ 和Python版本NPU输出问题

zengwubin

中级会员

积分
274
楼主
发表于 2020-4-11 18:04:44    查看: 7249|回复: 3 | [复制链接]    打印 | 只看该作者
YOLO-V3-tiny模型:C++:float* output_0=(float*)rknn_outputs[0].buf;
        float* output_1=(float*)rknn_outputs[1].buf;
        printf("\n size1 %d \n", rknn_outputs[0].size);  //打印输出:12168
        printf("\n size2 %d \n", rknn_outputs[1].size);  //打印输出:48672

python: boxes, boxes2 = rknn.inference(inputs=[imge])
             print(boxes)    #打印输出:(1, 18, 18, 13)
             print(boxes2)  #打印输出:(1, 18, 26, 26)

本人对C++ 不很熟,请讲下C++输出来的是什么??
回复

使用道具 举报

jefferyzhang

版主

积分
13580
沙发
发表于 2020-4-11 20:44:57 | 只看该作者
需要自己reshape成目标纬度
回复

使用道具 举报

zengwubin

中级会员

积分
274
板凳
 楼主| 发表于 2020-4-13 10:03:21 | 只看该作者
jefferyzhang 发表于 2020-4-11 20:44
需要自己reshape成目标纬度

但是我打印出来的size对应不上reshape的维度,请看上面
回复

使用道具 举报

jefferyzhang

版主

积分
13580
地板
发表于 2020-4-13 10:04:55 | 只看该作者
zengwubin 发表于 2020-4-13 10:03
但是我打印出来的size对应不上reshape的维度,请看上面

核对下api和toolkit和drv版本是否一致
回复

使用道具 举报

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

本版积分规则

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


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