专栏算法工具链输出能否保留NHWC

输出能否保留NHWC

已解决
HandsomeChen2024-02-05
105
2

模型的输入输出是NCHW，但是观察转换的中间产物xxx_quantized_model.onnx发现在BPU上是以NHWC进行计算的，最后输出的时候transpose改为NCHW。请问如何删除这个transpose，直接以NHWC作为输出？

1. 芯片型号：J5

2. 天工开物开发包 OpenExplorer 版本：例如 J5_OE_1.1.62

3. 问题定位：模型转换

4. 问题具体描述：请提供运行的命令、报错信息，如果可以的话，可以在附件中提供模型技术支持进行复现

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

    建议您参考地平线工具链在线手册：https://developer.horizon.cc/api/v1/fileData/horizon_j5_open_explorer_cn_doc/oe_mapper/source/ptq/ptq_tool/hb_model_modifier.html

    中的此章节：

    2024-02-05
    0
    1
    • Huanghui回复GOYUWJW:

      请问该问题是否已满足需求

      2024-02-18
      0