整理不易,如果觉得有用,记得点赞收藏和分享哦

1. 下载转换需要的代码文件

在下面地址下载代码文件

https://gitee.com/liangjiaxi2019/pytorch-YOLOv4

2. 安装onnxruntime

这里使用的是1.9.0版本的onnxruntime

pip3 install onnxruntime==1.9.0

3. 准备模型以及配置文件

在项目的data目录下准备转换需要的文件

需要模型架构的cfg文件,表示类别names文件,模型的权重weights文件以及一张测试图片

4.修改代码配置

修改demo_onnx.py文件
第56行写上当前模型预测的类别数目, 修改64行,改成实际模型的类别names文件

5. 运行转换

运行下面命令进行转换
其中最后一个参数是batchsize,这里以后模型推理是利用单张图输入,所以设成1

python3 demo_onnx.py data/xxxx.cfg data/xxxxx.weights data/testimage.jpg 1

6. 转换ONNX结果

转换完成以后,会生成onnx模型以及通过onnx模型预测的图片结果

darknet cpp weights模型转换成ONNX模型相关推荐

将训练好的pytorch模型的pth文件转换成onnx模型（亲测成功）
将训练好的pytorch模型的pth文件转换成onnx模型(亲测成功) 模型转换声明:本文原创,未经许可严禁转载,原文地址https://blog.csdn.net/hutao1030813002/ ...
【地平线开发板模型转换】将pytorch生成的onnx模型转换成.bin模型
文章目录 1 获取onnx模型 2 启动docker容器 3 onnx模型检查 3.1 为什么要检查? 3.2 如何操作 4 图像数据预处理 4.1 一些问题的思考 4.2 图片挑选与放置 4.2 使 ...
将Hugging Face模型转换成LibTorch模型
Hugging Face的模型以waifu-diffusion模型为例,给出的实现一般是基于diffuser库,示例代码如下: import torch from torch import auto ...
assimp批量转模型_IGS模型批量转换成STL模型
背景:做配载仪时需要批量的将IGS模型转换成STL模型用到的方法:方法1:使用3D-tool或者其他工具进行转换方法2:通过python使用FreeCAD的接口批量转换(当时师弟写的程序) Fre ...
pytorch模型(.pt)转onnx模型(.onnx)的方法详解(1)
1. pytorch模型转换到onnx模型 2.运行onnx模型 3.比对onnx模型和pytorch模型的输出结果我这里重点是第一点和第二点,第三部分比较容易首先你要安装依赖库:onnx ...
er图转换成关系模型的例题_有关数据库系统的练习题 E-R图的关系画图转换，，急需谢谢了...
展开全部你看62616964757a686964616fe58685e5aeb931333332643239下下边的例子,你的问题就可以解决了. 设某商业集团数据库中有三个实体集.一是"商 ...
Unity将内部模型转换成stl格式模型，用于3D打印机进行打印
本章我们一起来看下怎样将unity中的fbx模型转成stl模型并且保存到本地. 原理:stl模型都是由三角面组成的,只要我们了解stl文件的格式,就能够轻松的将fbx模型转换成stl. 1.先获取到f ...
现代控制理论的matlab上机实验将状态空间模型转换成传递函数模型（便于求各种响应）
现代控制理论用matlab将状态空间模型转换成传递函数模型(便于求各种响应) 例:matlab程序如下 A=[-21,19,-20;19,-21,20;40,-40,-40]; B=[0,1,2]' ...
sketchup 图片转模型_不用CAD描图迅速将图片转换成su模型
原标题:不用CAD描图迅速将图片转换成su模型首先上图,说明一下大概的流程: 1.[ps阶段]首先选取一副木雕图片,背景为单色最佳.如果不是单色,则需要ps一下,将背景转换成白色,这个过程使用魔术棒 ...

darknet cpp weights模型转换成ONNX模型

整理不易,如果觉得有用,记得点赞收藏和分享哦

1. 下载转换需要的代码文件

2. 安装onnxruntime

3. 准备模型以及配置文件

4.修改代码配置

5. 运行转换

6. 转换ONNX结果

darknet cpp weights模型转换成ONNX模型相关推荐

最新文章

热门文章