Skip to content

为什么rknpu和rknpu2的output_attr.dim排序是相反的? #57

@SpiceGL

Description

@SpiceGL

我在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上这种有点反直觉的情况我有些不太理解。

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions