1、将labelimg标注的xml、jpg文件分别放置在如图所示目录:
2、运行voc2yolo.py文件,转换数据为yolo格式
脚本下载地址:https://download.youkuaiyun.com/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文件需要修改的地方