专栏算法工具链XJ3的OE包能够支持5维输入

XJ3的OE包能够支持5维输入

已解决
gifbot5102023-03-21
37
3

你好!X3上转换模型生成bin文件,遇到输入shape问题:

  1. 系统软件版本:x3_ubuntu_v1.1.0

  2. 问题涉及的技术领域: 其他(算法应用)

  3. 问题描述:转换模型的过程中,hb_mapper是否支持5维输入(S×B×C×H×W)

  4. 复现概率:(必现) 已进行的排查措施、分析及结果:

onnx模型,包含两个输入。其中一个输入为5维。配置model_config.yaml文件,将input_type_rt、input_layout_rt、input_type_train、input_layout_train、input_shape改为相应数值。转换模型过程中,终端提示:

2023-03-20 21:52:04,844 INFO Model has 2 inputs according to model file
2023-03-20 21:52:04,844 INFO Input shape [**, **, **, **, **] has length: 5, make sure it is a featuremap input
根据《旭日X3派用户手册》第六章 转换内部过程解读 中的描述,输入feature只支持4维。

请问后续(X3后续OE包/J5平台)是否有支持5维输入的支持,或现版本(OE_2.4.2)是否有其他转换方法从而支持5维输入?

算法工具链
评论2
0/1000
  • 颜值即正义
    Lv.2

    你好,目前XJ3和J5均支持5维输入,对应的type需要配置为featuremap,layout可以配置为NHWC,会自动跳过layout

    2023-03-21
    2
    1
    • gifbot510回复颜值即正义:

      模型已转换成功,感谢!

      2023-03-21
      0
  • 颜值即正义
    Lv.2
    2023-04-24
    0
    0