使用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文件需要修改的地方

版权声明：本文为CSDN博主「雪地(>^ω^<)」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
原文链接：https://blog.csdn.net/qq_36563273/article/details/113606761