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
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