将pt文件转换成onnx文件

将pt文件转化为onnx文件

1 9 8 3 ?

3126人浏览 · 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社区

更多推荐

Vue3 现实世界的 Web 应用构建指南（二）

为了检索我们的锻炼，我们将在健身商店中添加一个新的方法。为此目的，我们还有一个视图准备就绪！让我们看看更新的健身商店文件：github.com/PacktPublishing/Building-Real-world-Web-Applications-with-Vue.js-3/blob/main/06.fitness/.notes/6.13-fitness.ts。如您所见，通过锻炼（第 15 行）