使用windows版yolov4训练自己的数据集

1、将labelimg标注的xml、jpg文件分别放置在如图所示目录:

在这里插入图片描述

2、运行voc2yolo.py文件,转换数据为yolo格式

脚本下载地址:https://download.csdn.net/download/qq_36563273/15008841
需要修改两处,一处为类别名称
在这里插入图片描述
一处为训练集占比
在这里插入图片描述
通过该脚本,会生成 2007_test.txt、2007_train.txt两个文件,ImageSets、labels两个文件夹
在这里插入图片描述

3、训练命令

在E:\Projects\darknet-v4\build\darknet\x64路径下:

darknet.exe detector train neibi\voc-neibi.data neibi\yolov4-tiny-neibi.cfg yolov4-tiny.conv.29 –map 

需要.data文件、.cfg文件、预训练权重yolov4-tiny.conv.29,.data文件里面会需要到.names文件
在这里插入图片描述

如果模型修改了,不能使用预训练权重,只需要在命令中去掉即可。

4、先验框聚类

darknet.exe detector calc_anchors battery\voc-battery.data -num_of_clusters 6 -width 640 -height 640

根据聚类出来的先验框,替换掉原来的anchor,可以提高精度。

5、cfg文件需要修改的地方

在这里插入图片描述
在这里插入图片描述

  • 2
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值