Toybrick

标题: 关于C++ 和Python版本NPU输出问题 [打印本页]

作者: zengwubin 时间: 2020-4-11 18:04
标题: 关于C++ 和Python版本NPU输出问题
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 时间: 2020-4-11 20:44
需要自己reshape成目标纬度

作者: zengwubin 时间: 2020-4-13 10:03

jefferyzhang 发表于 2020-4-11 20:44
需要自己reshape成目标纬度

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

作者: jefferyzhang 时间: 2020-4-13 10:04

zengwubin 发表于 2020-4-13 10:03
但是我打印出来的size对应不上reshape的维度，请看上面

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

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