pt权重转onnx记录

方法:

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

export.py

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

pt权重转onnx记录_第1张图片

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 

你可能感兴趣的:(DeepLearning,人工智能,目标检测,YOLO)