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

你好!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维输入?

感谢您使用地平线芯片算法工具链,最近我们在收集大家的满意度反馈,欢迎您填写问卷,详细情况可见:https://developer.horizon.ai/forumDetail/146177053698464782

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

模型已转换成功,感谢!