swintransformer_opset11.onnx模型转换报错

已解决

我爱邢宝宝i2024-02-29

105

1. 芯片型号：J5

2. 天工开物开发包 OpenExplorer 版本：j5_open_explorer_v1.1.68-py38_20231014

3. 问题定位：swintransformer_opset11.onnx模型转换

4. 问题具体描述：ERROR-OCCUR-DURING {horizon_nn.build_onnx} ***, error message: The input model is an invalid model(onnx shape_inference or onnx model_check failed). Error message:Node (/swin/encoder/layers.0/blocks.0/Unsqueeze) has input size 2 not in range [min=1, max=1]

算法工具链

征程5

0/1000

kotei左文亮
Lv.3
依据您发的信息，这个模型检查就有问题啊，能否把onnx文件发出来啊？
2024-03-01
0
3
- 我爱邢宝宝i回复kotei左文亮:
  您好，这种是有算子不支持吗
  2024-03-02
  0
  回复
- 我爱邢宝宝i回复kotei左文亮:
  链接：https://pan.baidu.com/s/1NNeiaeu34sPfAwIOMCgreg 提取码：BJ40
  2024-03-04
  0
  回复
- 我爱邢宝宝i回复kotei左文亮:
  您好，烦请空闲时间回复一下
  2024-03-07
  0
  回复
kotei左文亮
Lv.3
ONNX模型检测的过程中与没有啥问题？
2024-03-01
0
1
- 我爱邢宝宝i回复kotei左文亮:
  ERROR *** ERROR-OCCUR-DURING {horizon_nn.build_onnx} ***, error message: The input model is an invalid model(onnx shape_inference or onnx model_check failed). Error message:Node (/swin/encoder/layers.0/blocks.0/Unsqueeze) has input size 2 not in range [min=1, max=1]. ==> Context: Bad node spec: input: "/swin/encoder/layers.0/blocks.0/Mod_1_output_0" input: "onnx::Unsqueeze_306" output: "/swin/encoder/layers.0/blocks.0/Unsqueeze_output_0" name: "/swin/encoder/layers.0/blocks.0/Unsqueeze" op_type: "Unsqueeze". Please check you input model again.
  2024-03-01
  0
  回复
我爱邢宝宝i
Lv.2
嗯嗯，主要看咱们介绍，测过SwinTransformer的性能，想着应该有对应的onnx

2024-03-11
1
0
kotei左文亮
Lv.3
已经在处理了，稍等~~
2024-03-07
0
0
kotei左文亮
Lv.3
OE68版本，使用onnx模型检测发现unsqueeze算子会被转换成Reshape，BPU约束如下：1. 该算子支持int16输入输出。2. 支持1-10维输入输出
初步解决思路是更换次算子的输入输出或者替换次算子，在重新生成onnx模型进行转化。
2024-03-07
0
0
我爱邢宝宝i
Lv.2
好的，咱们那边有swintransformer的onnx模型吗，能否方便提供一下，谢谢

2024-03-09
0
0
kotei左文亮
Lv.3
目前68版本的OE包里面的样例是没有这个网络的onnx文件的，后续有没有还不确定，建议您先自己训练这个网络，然后转onnx试一下。
2024-03-11
0
0