ITPub博客

首页 > 人工智能 > 人工智能 > 【玩转香橙派Orange Pi】关于pytorch模型转化为om过程中遇到的问题

【玩转香橙派Orange Pi】关于pytorch模型转化为om过程中遇到的问题

原创 人工智能 作者:zzb_3359 时间:2024-05-10 19:45:17 0 删除 编辑

1 自有项目基于yolov5开发的,pytorch是1.12版本的,模型已经训练完成,移植到香橙派Aipro 需要用到我的onnx模型,但是我手里的是pt模型,所以先进行第一步的onnx转换。

2 转换报错,报错为opset version:17的错误。说明我转换时候用的opset版本不对。


3 问题解决办法。

原文链接地址: https://blog.csdn.net/qq_43318374/article/details/120195550

4 查询本地的onnx版本

我的版本查到的是1.15.0   那么对应的opset应该设置为15以下,不能大于15就可以转换成功。

来自 “ ITPUB博客 ” ,链接:https://blog.itpub.net/70038465/viewspace-3015014/ ,如需转载,请注明出处,否则将追究法律责任。

上一篇: 没有了~
请登录后发表评论 登录
全部评论

注册时间:2024-05-08

  • 博文量
    6
  • 访问量
    4951