专栏算法工具链关于J5 BPU SDK手册中的数据Layout问题

关于J5 BPU SDK手册中的数据Layout问题

已解决
默认823682024-09-09
37
1

OE手册中在BPU SDK API中写的数据layout,

介绍了两种数据排布: NHWC_NATIVE 和 NCHW_NATIVE ;

我用AI_Benchmark尝试接入自己的PTQ模型,输出为5维,例:[1,3,80,80,6];

请问数据排布是怎么样的,是否有相应文档?

在自定义的Postprocess_method.cc中应该怎么对其张量?

算法工具链
评论1
0/1000
  • Pipeline
    Lv.2

    NCHW和NHWC针对四维数据,超过就不适用,按照输入featuremap的shape解析就行

    2024-09-09
    0
    0