yolov5 segmented error: Error in predict_shapes: shapes (3,31) and (32,25600) not aligned: 31 (dim 1) != 32 (dim 0)

[2375963934a]

我将我的yolov5m-seg生成的pt文件转换成onnx格式加载模型进去并修改了yaml文件，但是运行的时候不进行标注 会卡在这里，然后就退出程序了请问该怎么解决

[CVHub520]

您好，请问源码运行下，贴下报错信息。另外请提供下您的配置文件截图以及onnx模型的输入输出节点信息。

[2375963934a]

这是配置文件截图

[CVHub520]

您好，感谢反馈。看起来貌似是你的路径没设置对。你可以保持上述配置文件的设置，然后新建一个文件夹，将你转换好的 onnx 文件和这份配置文件放置到新建文件夹下。最后再通过界面导入自定义模型选择此配置文件即可。

[2375963934a]

![Uploading image.png…]() 还是不会进行标注

[2375963934a]

显示完成推理但是没有进行标注也没有生成标签文件

[CVHub520]

请问内置的 yolov5-seg 能正常运行吗？

[2375963934a]

![Uploading image.png…]() 内置的直接闪退了

[2375963934a]

[CVHub520]

我看你这模型还没下载成功吧？

[2375963934a]

这里显示模型加载成功但是点运行就会闪退

[CVHub520]

可以把模型权重、配置文件和一张确保能推理出目标的图片打包发送至邮箱：cv_hub@163.com

[dahaigui]

我这里也遇到了类似问题， 用官方的代码可以实现预测。

[CVHub520]

@dahaigui，Hi，感谢反馈！ 如果方便的化，麻烦将模型权重、配置文件和一张确保能推理出目标的图片打包发送至邮箱：cv_hub@163.com

[dahaigui]

@dahaigui，Hi，感谢反馈！ 如果方便的化，麻烦将模型权重、配置文件和一张确保能推理出目标的图片打包发送至邮箱：cv_hub@163.com

已经发送到您的邮箱了，麻烦了！

[CVHub520]

@dahaigui，Hi，感谢反馈！ 如果方便的化，麻烦将模型权重、配置文件和一张确保能推理出目标的图片打包发送至邮箱：cv_hub@163.com

已经发送到您的邮箱了，麻烦了！

收到，谢谢！

[2375963934a]

请问这个问题解决了吗

[CVHub520]

请问这个问题解决了吗

如果方便的话，麻烦提供下模型权重、配置文件和一张服务器模型能推理出目标的图片打包发送至邮箱：cv_hub@163.com 以更好的定位问题。

[2375963934a]

已经发送到您的邮箱了，麻烦了！

[CVHub520]

您好，麻烦在运行前仔细检查自己的配置文件与实际模型是否相符，你的标签类别信息和模型文件信息明显对应不上，请不要自己随便发挥。

type: yolov5_seg
name: yolov5s-seg-r20231027
display_name: YOLOv5s-Seg Ultralytics
model_path: best.onnx
nms_threshold: 0.45
confidence_threshold: 0.25
classes:
  - zase
  - feibian
  - bapi
  - queliao

[2375963934a]

在我的电脑上运行就直接闪退了呀

[2375963934a]

卡死在这然后闪退 yaml配置文件是这样子的 没有报错直接闪退

[CVHub520]

@2375963934a，您好，建议使用 CPU 环境运行，有可能是 CUDA 版本不匹配。

[2375963934a]

我的cuda是12.1的成功运行了一张图片然后就闪退了，是什么原因

[CVHub520]

我的cuda是12.1的成功运行了一张图片然后就闪退了，是什么原因

可以贴下具体的报错信息

[2375963934a]

没有报错信息就是直接闪退了

[CVHub520]

没有报错信息就是直接闪退了

那应该就是版本没安装对，你这边如果不会配环境的话建议还是用cpu，环境搭建是比较麻烦，具体你可以查看官方文档。