在这里插入图片描述

将pt文件转换成onnx文件

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

python export.py --include torchscript onnx

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

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

在这里插入图片描述

Logo
2048 AI社区

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

更多推荐

cover

基于LLM大模型的股票基金周预测Agent

avatar 2048 AI社区
cover

‌AI在自动化测试中的角色：助手还是主导

avatar 2048 AI社区

Flutter框架跨平台鸿蒙开发——宿舍报修APP的开发流程

本文介绍了基于Flutter框架开发跨平台鸿蒙宿舍报修APP的完整流程。该应用采用Flutter 3.6.2和Dart 3.6.2技术栈，实现了学生报修申请、报修管理、状态更新等功能。开发过程包括需求分析、分层架构设计、数据库建模和界面开发，采用Material Design风格，构建了包含主页面、添加报修页面和详情页面的完整应用。核心功能通过RepairRequest模型类和DatabaseHe

avatar 2048 AI社区