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

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 --batch-size 16

weights：指你训练好的权重文件，看你放在哪个文件夹中。
batch-size:指你训练的时候的batch-size大小，export.py默认是1，需要根据你训练的时候修改。

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