将pt文件转换成onnx文件

将pt文件转化为onnx文件

1 9 8 3 ?

3286人浏览 · 2023-11-04 09:13:47

1 9 8 3 ? · 2023-11-04 09:13:47 发布

在这里插入图片描述

将pt文件转换成onnx文件

打开yolov5自带的export.py文件更改weights信息
在这里插入图片描述
更改完成后在终端输入

python export.py --include torchscript onnx

如此即可实现pt到onnx的转换,路径在pt文件同级
在这里插入图片描述
如果要生成engine（用于tensorRT加速）格式则输入

python export.py --include engine --device 0

在这里插入图片描述

2048 AI社区

有“AI”的1024 = 2048，欢迎大家加入2048 AI社区

更多推荐

UFT 对象定位的问题及技巧（含正则实战 & 无代码对比）

本文探讨了UFT自动化测试中对象定位的核心问题及解决方案。作者指出动态ID、层级变动和多语言适配是导致定位失败的三大主因，提出使用正则表达式匹配、描述性编程和层级缩减策略等技巧提升稳定性。针对金融系统常见的Grid控件，文章分享了特殊处理方法，并总结了对象定位的四大稳定原则。最后对比了传统UFT脚本与无代码自动化系统(MARS)的差异，强调语义封装和对象模型库的重要性。文章认为，在AI时代，稳定的