Failed to execute script 'pegasus' due to unhandled exception!

[plustree]

Failed to execute script 'pegasus' due to unhandled exception!

[WhycanService]

@plustree 有没有截图

[plustree]

@whycanservice pegasus import onnx --model yolov5n-sim.onnx --output-data yolov5n-sim.data --output-model yolov5n-sim.json --outputs "350 498 646"

......
W Call onnx.optimizer.optimize fail, skip optimize
I Current ONNX Model use ir_version 8 opset_version 17
Traceback (most recent call last):
File "pegasus.py", line 131, in <module>
File "pegasus.py", line 112, in main
File "acuitylib/app/importer/commands.py", line 248, in execute
File "acuitylib/vsi_nn.py", line 125, in load_onnx
File "acuitylib/app/importer/import_onnx.py", line 117, in run
File "acuitylib/converter/onnx/convert_onnx.py", line 62, in init
File "acuitylib/converter/onnx/convert_onnx.py", line 90, in __build_specified_subgraph
File "acuitylib/converter/onnx/onnx_util.py", line 400, in build_subgraph
File "acuitylib/converter/onnx/onnx_util.py", line 400, in <listcomp>
KeyError: '350'
[******] Failed to execute script 'pegasus' due to unhandled exception!

[plustree]

@whycanservice https://v853.docs.aw-ol.com/npu/npu_yolov5/

[plustree]

@whycanservice yolov5$ pegasus import onnx --model yolov5n-sim.onnx --output-data yolov5n-sim.data --output-model yolov5n-sim.json --outputs "350 498 646"

[plustree]

@whycanservice 您好，这个问题是参数因为没有写对吗

[WhycanService]

@plustree 【Genin】大家碰到过这个问题嘛 pegasus的
https://bbs.aw-ol.com/topic/3064/share/6

[misay]

同样的问题，不知道你解决了吗，希望把后续贴上来

[masterlin]

@plustree 您好，请问这个要怎么解决呀，我用yolov5-7.0的模型会有这个问题，6.0没有，但是需要用到7.0的语义分割模型。想请教下如何解决，谢谢大佬

[masterlin]

@whycanservice 您好，请问这个要怎么解决呀，我用yolov5-7.0的模型会有这个问题，6.0没有，但是需要用到7.0的语义分割模型。想请教下如何解决，谢谢大佬

[masterlin]

@misay 您好，请问这个要怎么解决呀，我用yolov5-7.0的模型会有这个问题，6.0没有，但是需要用到7.0的语义分割模型。想请教下如何解决，谢谢大佬