pt权重文件转onnx详解（记录）

pt权重转onnx记录

方法

1.一般YOLOv3和YOLOv5 会自带一个权重pt转onnx模型的代码

export.py

export.py在models文件夹中

2.打开export.py ，我们运行该文件可以通过命令行运行，对照修改argument参数即可

3.运行命令行如下：

python ./models/export.py --weights ./models/best.pt

weights：指你训练好的权重文件，看你放在哪个文件夹中。

4.导出的onnx文件就会在你best.pt的相同目录当中，你不修改–include参数的话，一般会出现三个文件，分别是torchscript, onnx, coreml。一般onnx和torchscript文件大小为你的pt文件的两倍大小，如果不是2倍大小，那可能就是转错了。

注意：不要进入models文件夹中运行export.py，要不然会报错误，找不到model