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

始不晚 · 发表于 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

安装完依赖库后,cd到yolo5的开源库同目录后,输入命令行:
python export.py --weights last.pt --include torchscript onnx

然后你就得到了onnx模型啦.

YOLO4教程:
【YOLO4模型转换】darknet-YOLO4模型转ONNX详细教程，附开源包

本贴附件:

游客，如果您要查看本帖隐藏内容请回复

我是新手 · 发表于 12 小时前

大佬牛逼牛逼超级牛逼

774172350 · 发表于 2025-6-13 02:17:31

6666666666666

zz8zz · 发表于 2025-5-31 22:36:10

kca · 发表于 2025-3-14 19:49:26

6666666666666

yyk80238 · 发表于 2025-3-11 17:29:22

wq838339083 · 发表于 2025-3-4 00:22:29

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

意难平 · 发表于 2025-1-31 23:24:52

学习学习爱你爱你

callgood · 发表于 2024-11-16 12:40:45

非常好,感谢

316andy · 发表于 2024-11-14 10:56:43

66666666666666

		自动登录	找回密码
密码			立即注册