递归火山软件开发平台

 找回密码
 立即注册

QQ登录

只需一步,快速开始

搜索
热搜: 火山 源码 类库
查看: 10270|回复: 112

[YOLO5模型转换]教你YOLO5模型pt转onnx详细教程,附文件

  [复制链接]

7

主题

89

帖子

616

积分

核心用户

Rank: 9Rank: 9Rank: 9

积分
616
QQ
发表于 2022-10-5 14:37:16 | 显示全部楼层 |阅读模式
本帖最后由 始不晚 于 2022-10-10 11:01 编辑

如果你已用yolo5训练出了pt模型,但想用火山CV类库调用,就必须转换为onnx
非常简单,如果Anaconda环境已安装,那么就一条命令即可.
如果未安装Anaconda,我给大家准备了推荐:
游客,如果您要查看本帖隐藏内容请回复

首先打开Anaconda,进入已创建的yolo环境.并安装依赖库:
pip install onnx coremltools onnx-simplifier
1.png
2.png
安装完依赖库后,cd到yolo5的开源库同目录后,输入命令行:
python export.py --weights last.pt --include torchscript onnx
3.png
然后你就得到了onnx模型啦.
4.png
YOLO4教程:
【YOLO4模型转换】darknet-YOLO4模型转ONNX详细教程,附开源包

本贴附件:
游客,如果您要查看本帖隐藏内容请回复



评分

参与人数 1金钱 +15 收起 理由
lu9944 + 15 如果是yolov5项目,项目需要使用6.0-6.2版.

查看全部评分

回复

使用道具 举报

8

主题

51

帖子

495

积分

核心用户

Rank: 9Rank: 9Rank: 9

积分
495
发表于 2022-10-5 14:52:49 | 显示全部楼层
我来了正式我需要的爱死你了群主:噜阿噜
回复

使用道具 举报

10

主题

138

帖子

3890

积分

核心用户

Rank: 9Rank: 9Rank: 9

积分
3890
发表于 2022-10-5 14:57:07 | 显示全部楼层
66666666666666
回复

使用道具 举报

1

主题

33

帖子

505

积分

高级会员

Rank: 4

积分
505
发表于 2022-10-5 15:09:20 | 显示全部楼层
学习学习
回复

使用道具 举报

0

主题

3

帖子

8

积分

新手上路

Rank: 1

积分
8
发表于 2022-10-5 15:16:53 | 显示全部楼层
如果您要查看本帖隐藏内容请回复
回复

使用道具 举报

28

主题

205

帖子

3227

积分

核心用户

Rank: 9Rank: 9Rank: 9

积分
3227
发表于 2022-10-5 15:26:21 | 显示全部楼层
666666
回复

使用道具 举报

1

主题

38

帖子

537

积分

核心用户

Rank: 9Rank: 9Rank: 9

积分
537
发表于 2022-10-5 15:33:39 | 显示全部楼层
学习学习
回复

使用道具 举报

5

主题

115

帖子

1199

积分

核心用户

Rank: 9Rank: 9Rank: 9

积分
1199
发表于 2022-10-5 15:35:14 | 显示全部楼层
!――
凝逸AI:nyav.net、Ai图像识别
回复

使用道具 举报

3

主题

29

帖子

603

积分

核心用户

Rank: 9Rank: 9Rank: 9

积分
603
发表于 2022-10-5 15:46:04 | 显示全部楼层
66666666666666
回复

使用道具 举报

1

主题

43

帖子

562

积分

核心用户

Rank: 9Rank: 9Rank: 9

积分
562
发表于 2022-10-5 15:53:26 | 显示全部楼层

66666666666666
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

QQ|Archiver|手机版|小黑屋|递归火山软件开发平台 ( 鄂ICP备18029190号 )

GMT+8, 2024-6-24 14:33 , Processed in 0.095185 second(s), 32 queries .

Powered by Discuz! X3.4

© 2001-2017 Comsenz Inc.

快速回复 返回顶部 返回列表