pt权重文件转onnx详解(记录)

pt权重转onnx记录

方法

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

export.py

export.py在models文件夹中

pt权重文件转onnx详解(记录)_第1张图片
2.打开export.py ,我们运行该文件可以通过命令行运行,对照修改argument参数即可
pt权重文件转onnx详解(记录)_第2张图片
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

你可能感兴趣的:(onnx端化,pytorch)