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

始不晚 · 发表于 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详细教程,附文件

本贴附件：

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

chenshui · 发表于 2022-10-5 11:36:22

谢谢大佬分享经验

xqnh520 · 发表于 2022-10-5 11:36:41

学习了????

Kyu · 发表于 2022-10-5 11:38:20

谢谢大佬分享经验

windkey · 发表于 2022-10-5 11:52:54

学习一下

675550115 · 发表于 2022-10-5 11:58:23

看看什么好东西！

君安光电科技 · 发表于 2022-10-5 12:00:53

好~好~好~非常好~大大的好~好就是好~就是好~是好~好~

凝逸AI · 发表于 2022-10-5 12:09:06

xxxxxxxx

2995455338 · 发表于 2022-10-5 12:31:20

谢谢大佬分享经验

zx896825328 · 发表于 2022-10-5 12:54:06

谢谢大佬分享经验

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