yolo3,模型训练流程

本文详细介绍了如何使用YOLOv3-SPP进行物体检测模型训练,包括下载并编译Darknet,准备数据集,修改配置文件,创建.names和.data文件,整理样本txt,以及运行训练命令的具体步骤。

训练前准备:

下载darknet_win版本,并用vs2015编译对用的gpu,cpu版本。

1,准备数据集

图片和边界矩形txt

边界矩形txt内容

类别从0开始

labelText = str(int(labstr)-1) + ' ' + str(x) + ' ' + str(y) + ' ' + str(w) + ' ' + str(h) + '\n'

x,y,w,h为目标的矩形中心(x,y),wh目标宽高

 

2,修改cfg文件

[convolutional]
size=1
stride=1
pad=1
filters=27                    修改为(cls+5)*3
activation=linear


[yolo]
mask = 0,1,2
anchors = 75,45, 87,56, 145,50, 171,136, 194,91, 240,116, 246,94, 278,53, 314,150           修改boxes的kmean
classes=4                修改类别

 

3,创建.names文件,内容为分类名字

4,创建.data文件

例如:

classes= 4
train  = trainxx.txt
valid  = testxx.txt
names = data/xx.names
backup = mod/xx

5,数据集

样本txt整理,如上面的trainxx.txt

6,运行命令

darknet detector train xx.data data/yolov3-spp-xx.cfg data/darknet53.conv.74

 

 

 

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值