Win10系统下使用pytorch1.11.0、onnx1.12.0转换yolov5cls转换onnx报错如下。

在这里插入图片描述
查找了相关资料后是由于pytorch对onnx算子的不支持

参考资料:
1、模型部署,https://zhuanlan.zhihu.com/p/498425043
2、onnx算子文档,https://github.com/onnx/onnx/blob/main/docs/Operators.md

Logo

GitCode 天启AI是一款由 GitCode 团队打造的智能助手,基于先进的LLM(大语言模型)与多智能体 Agent 技术构建,致力于为用户提供高效、智能、多模态的创作与开发支持。它不仅支持自然语言对话,还具备处理文件、生成 PPT、撰写分析报告、开发 Web 应用等多项能力,真正做到“一句话,让 Al帮你完成复杂任务”。

更多推荐