火山软件开发平台

 找回密码
 立即注册

QQ登录

只需一步,快速开始

搜索
热搜: 火山 源码 类库
楼主: 始不晚
打印 上一主题 下一主题

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

  [复制链接]

7

主题

89

帖子

604

积分

核心用户

Rank: 9Rank: 9Rank: 9

积分
604
QQ
跳转到指定楼层
楼主
发表于 2022-10-5 11:09:14 | 显示全部楼层 |只看大图 回帖奖励 |倒序浏览 |阅读模式
本帖最后由 始不晚 于 2022-10-8 22:55 编辑

!!!!YOLO4不需要转onnx了,可直接使用DNN调用!!!
------------------------------------------------------------------------
OpenCV类库千呼万唤始出来,那么有了利器,我们就得出鞘。
在火山发布该类库之前,很多人都在用易语言大佬发出的工具训练自己的YOLO4或YOLO5模型,但是,YOLO4模型为

YOLO5的则是.pt
今天说下YOLO4,如darknet_last.weights的权重模型,那么火山封装为调用onnx

我们如何调用训练好的模型呢?
首先我么需要安装Anaconda,我给大家准备了:
游客,如果您要查看本帖隐藏内容请回复

第一步是安装Anaconda,第二步是创建虚拟环境和安装对应的引用库
------------------------------------------------------------------------
在这里,我就当大家已经安装好了环境。
转换需要大佬的开源项目,我已打包上传。
解压后,我们打开Anaconda,并进入到自己创建的yolo环境。如图

然后cd到这个开源包的解压路径:

然后输入命令:
python demo_darknet2onnx.py ./res3/darknet.cfg ./res3/labels.names ./res3/darknet_last.weights ./res3/RC.png 1
请根据自身情况更改,输入后回车。因为好多人是用darknet-yolo4训练模型,所以用这个demo_darknet2onnx.py
如果你是用的pytorch-yolo4训练的模型,就换成demo_pytorch2onnx.py即可。

经过几秒钟,你就会在该开源包的同目录获得ONNX文件。

然后你就能在火山CV机器视觉AI类库中调用啦。
----------------------------------------------------------------
YOLO5也很简单哦。[YOLO5模型转换]教你YOLO5模型pt转onnx详细教程,附文件


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








回复

使用道具 举报

7

主题

89

帖子

604

积分

核心用户

Rank: 9Rank: 9Rank: 9

积分
604
QQ
沙发
 楼主| 发表于 2022-10-5 12:58:06 | 显示全部楼层
zx896825328 发表于 2022-10-5 12:56
有个问题 载入模型是路径  能字节集载入   防止被偷

OK,反馈一下
回复

使用道具 举报

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

本版积分规则

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

GMT+8, 2024-5-4 13:33 , Processed in 0.092181 second(s), 22 queries .

Powered by Discuz! X3.4

© 2001-2017 Comsenz Inc.

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