pytorch训练好的YOLOv5模型转paddle

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=(保存生成好模型的地址)