我在rv1126上使用的rknpu和rknn-toolkit版本为1.7.1,在rk3588上使用的rknpu2和rknn-toolkit2版本为1.4.0。
使用同一个模型(YOLOv7)进行测试,使用通用API进行推理。
在rv1126上,rknn_outputs_get获取到的输出outputs的buf数据的排列顺序确实是NCHW(1,3,400,7),但对应的output_attr.dims值按顺序是(7,400,3,1)。
而在rk3588上buf数据的顺序和output_attr.dims的顺序是一致的,都是(1,3,400,7)。
rv1126上这种有点反直觉的情况我有些不太理解。