pt权重转onnx记录

方法:

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

export.py

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

转自：

pt权重文件转onnx详解（记录）_pt转onnx-CSDN博客文章浏览阅读1.3w次，点赞9次，收藏58次。pt权重转onnx记录方法1.一般YOLOv3和YOLOv5 会自带一个权重pt转onnx模型的代码export.pyexport.py在models文件夹中2.打开export.py ，我们运行该文件可以通过命令行运行，对照修改argument参数即可3.运行命令行如下：python ./models/export.py --weights ./models/best.pt --batch-size 16weights：指你训练好的权重文件，看你放在哪个文件夹中。batch-_pt转onnxhttps://blog.csdn.net/weixin_44312422/article/details/121510796