始不晚 发表于 2022-10-5 14:37:16

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

本帖最后由 始不晚 于 2022-10-10 11:01 编辑

如果你已用yolo5训练出了pt模型,但想用火山CV类库调用,就必须转换为onnx
非常简单,如果Anaconda环境已安装,那么就一条命令即可.
如果未安装Anaconda,我给大家准备了推荐:
**** Hidden Message *****
首先打开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详细教程，附开源包

本贴附件:
**** Hidden Message *****


q1015133989 发表于 2022-10-5 14:52:49

我来了正式我需要的爱死你了群主:噜阿噜

2995455338 发表于 2022-10-5 14:57:07

66666666666666

邪lei 发表于 2022-10-5 15:09:20

学习学习

回忆那浅笑 发表于 2022-10-5 15:16:53

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

666 发表于 2022-10-5 15:26:21

666666

gzsam 发表于 2022-10-5 15:33:39

学习学习

凝逸AI 发表于 2022-10-5 15:35:14

！――

fightboy 发表于 2022-10-5 15:46:04

66666666666666

pkwwfpkwwf 发表于 2022-10-5 15:53:26


66666666666666
页: [1] 2 3 4 5 6 7 8 9 10
查看完整版本: [YOLO5模型转换]教你YOLO5模型pt转onnx详细教程,附文件