lableimg闪退_yolov4——训练自己的数据集(完整版)

好了，多的也不说，直接开始吧！

1.下载预训练的好权重文件 yolov4.conv.137,放在build/darknet/x64/下

2.创建配置文件：在darknet-master/cfg/下，创建yolo-obj_fre.cfg：

这个文件里面的内容跟cfg/yolov4-custom.cfg类似，只是有几个地方需要修改

1) batch=8 #依据你电脑的显存大小而定，尽量设大一点2) subdivisions=64 # 这里最好设为64，不然训练的时候会报错3) max_batches=8000 #classes*20004) steps=6400,7200 #max_batches*0.8,max_batches*0.95) 修改3个[yolo]下classes=4 #你要训练的类别数6) 修改3个[yolo]上面filters=27 #filters=(classes + 5)x3，注意只修改每个[yolo]上面最后一个conv的filters

3.在build\darknet\x64\data\下，创建fre.names文件，其中每一行写上一个你要训练的一个类别，eg：applebicklycellbag

4.在build\darknet\x64\data\下，创建fre.data文件:

classes= 4train = build/darknet/x64/data/train.txt # 这个下面说，train.txt里面存放着训练图片的存放路径valid = build/darknet/x64/data/test.txt #names = build/darknet/x64/data/fre.names # 步骤3创建的fre.names