【Genin】大家碰到过这个问题嘛 pegasus的

[w2836833]

[WhycanService]

麻烦完整截图。。。

[fawen]

@w2836833 这个问题解决了吗

[WhycanService]

onnx版本对不上，onnx建议对应上onnx-sim的版本

[fawen]

@fawen pegasus import onnx --model yolov5s-sim.onnx --output-data yolov5s-sim.data --output-model yolov5s-sim.json

[WhycanService]

可以参考这篇文章 https://blog.csdn.net/tugouxp/article/details/125570628

[plustree]

@fawen ValueError: operands could not be broadcast together with shapes (1,3,80,80,0) (1,3,80,80,2)
[495133] Failed to execute script 'pegasus' due to unhandled exception!

[misay]

@plustree 找到解决方法了，报错原因是onnx的三个输出的name一定要和指令厘米的name对应上

[feidedao]

@w2836833
yolov5s-sim-v60.onnx 用这个6.0版本的，对应论坛的教程

[masterlin]

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