rknpu
rknpu copied to clipboard
为什么rknpu和rknpu2的output_attr.dim排序是相反的?
我在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上这种有点反直觉的情况我有些不太理解。