瑞客论坛

 找回密码
 立即注册
查看: 7320|回复: 42

[经验求助] yolo如何导出.weights和.cfg

[复制链接]

金币2362  第1024名

6

主题

1173

回帖

6728

积分

论坛元老

Rank: 8Rank: 8

威望
2619
贡献
1746
热心值
1
金币
2362
注册时间
2023-12-7
发表于 2024-1-16 17:55 | 显示全部楼层 |阅读模式
10金币



各位老师好,最近新接触一个opencv库中的函数cv::dnn::readnet()函数;        参数1是yolo训练好的模型.weights后缀文件,
        参数2是超参数.cfg后缀的文件。
如何才能得到这2个后缀的文件呢(*.weigthts和*.cfg),或者其他的也行,因为yolov8训练好的都是*.pt和*.yaml配套。网上还可以发现能导出*.onnx,但是第二个参数怎么整呀?
如何导出一套来呢?只要能用上readnet函数就可以。



官网解释:
Net cv::dnn::readNet
(const String & model,
const String & config = "",
const String & framework = ""
)
Python:
cv.dnn.readNet(model[, config[, framework]]) ->retval
cv.dnn.readNet(framework, bufferModel[, bufferConfig]) ->retval
Parameters
[in]modelBinary file contains trained weights. The following file extensions are expected for models from different frameworks:
[in]configText file contains network configuration. It could be a file with the following extensions:
[in]frameworkExplicit framework name tag to determine a format.


回复

使用道具 举报

金币2031  第1208名

2

主题

3094

回帖

1万

积分

论坛元老

Rank: 8Rank: 8

威望
7453
贡献
5012
热心值
0
金币
2031
注册时间
2021-8-26
发表于 2024-1-16 21:55 | 显示全部楼层
回复

使用道具 举报

金币25  第13868名

5

主题

1334

回帖

5868

积分

论坛元老

Rank: 8Rank: 8

威望
2987
贡献
2855
热心值
1
金币
25
注册时间
2019-11-28
发表于 2024-1-16 22:46 | 显示全部楼层
回复

使用道具 举报

金币5161  第362名

0

主题

2768

回帖

1万

积分

论坛元老

Rank: 8Rank: 8

威望
8286
贡献
5706
热心值
0
金币
5161
注册时间
2021-1-30
发表于 2024-1-17 07:00 | 显示全部楼层
666666
回复

使用道具 举报

金币5161  第362名

0

主题

2768

回帖

1万

积分

论坛元老

Rank: 8Rank: 8

威望
8286
贡献
5706
热心值
0
金币
5161
注册时间
2021-1-30
发表于 2024-1-17 07:01 | 显示全部楼层
666666
回复

使用道具 举报

金币5557  第319名

0

主题

2522

回帖

1万

积分

论坛元老

Rank: 8Rank: 8

威望
7812
贡献
5398
热心值
0
金币
5557
注册时间
2021-12-30
发表于 2024-1-17 07:01 | 显示全部楼层
666666
回复

使用道具 举报

金币5557  第319名

0

主题

2522

回帖

1万

积分

论坛元老

Rank: 8Rank: 8

威望
7812
贡献
5398
热心值
0
金币
5557
注册时间
2021-12-30
发表于 2024-1-17 07:02 | 显示全部楼层
666666
回复

使用道具 举报

金币5156  第363名

0

主题

2639

回帖

1万

积分

论坛元老

Rank: 8Rank: 8

威望
8047
贡献
5547
热心值
0
金币
5156
注册时间
2021-12-14
发表于 2024-1-17 07:02 | 显示全部楼层
666666
回复

使用道具 举报

金币5156  第363名

0

主题

2639

回帖

1万

积分

论坛元老

Rank: 8Rank: 8

威望
8047
贡献
5547
热心值
0
金币
5156
注册时间
2021-12-14
发表于 2024-1-17 07:03 | 显示全部楼层
666666
回复

使用道具 举报

金币1077  第2065名

5

主题

1225

回帖

1万

积分

论坛元老

Rank: 8Rank: 8

威望
4566
贡献
4768
热心值
0
金币
1077
注册时间
2021-2-27
发表于 2024-1-17 10:05 | 显示全部楼层
在yolov5的export.py代码中添加:
from models.yolo import Model
model=Model(‘yolov5s.yaml’)
model.load_state_dict(torch.load(‘yolov5s.pt’)[‘model’].state_dict())
torch.save(model.state_dict(),‘yolov5s.weights’)
回复

使用道具 举报

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

本版积分规则

Archiver|手机版|小黑屋|瑞客论坛 |网站地图

GMT+8, 2026-3-25 16:03

Powered by Discuz! X3.4

© 2001-2023 Discuz! Team.

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