当前位置:   article > 正文

pytorch训练好的YOLOv5模型转paddle_onnx export failure: no module named 'onnx

onnx export failure: no module named 'onnx

1.best.pt转best.onnx

(1)准备工作,z在torch环境下,下载X2paddle(用于将其他架构转化为paddle)

pip install x2paddle -i https://simple.baidu.com/pypi/simple

(2)训练好的best.pt文件转化为.onnx文件

2.1找到models/export.py

 2.2调参(为避免报错,建议使用绝对地址)

-- weight:在default里面填写“best.pt地址”

-- device:如果你的torch是cpu版本,那么default里面填写“cpu”,如果是gpu版本,那么填写“0”

2.3运行即可

2.4获得best.onnx,通常存放在best,pt那一级(yolov5-5.0\runs\train\exp\best.onnx)

 2.5可能的报错:

2.5.1 ONNX export failure: No module named 'onnx'

很简单,直接在pycharm终端 

pip install onnx

如果显示已经安装(already)那么可能是版本不对

pip install onnx==1.7.0

2.5.2 CoreML export failure: No module named 'coremltools'

pip install coremltools

2.best.onnx转paddlepaddle框架

2.1 只需要一行代码(下图是实例,具体地址还需更改,注意地址不要有中文),仍然是在pycharm终端运行

x2paddle --framework=onnx --model=onnx_model.onnx --save_dir=pd_model 
x2paddle --framework=onnx --model=(存放best.onnx的地址,最好绝对地址) --save_dir=(保存生成好模型的地址) 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/Monodyee/article/detail/214981?site
推荐阅读
相关标签
  

闽ICP备14008679号