1.修改cfg文件夹中的voc.data文件:
【目的在于确定你需要做几个类别的物体检测,也就是class=1还是2还是其他。我这里检测的是2类,所以class=2,并且指明训练集和验证集的路径】
classes= 2
train = /home/sq123/data/2007_train.txt
valid = /home/sq123/data/2007_test.txt
names = data/voc.names
backup = backup
注意:class 为训练的类别数
train为训练集train.txt,其中包含训练样本的具体路径(具体生成方式见2)
valid为验证集val.txt,其中包含验证样本的具体路径(具体生成方式见2)
backup为weights的存储位置(可以不修改,修改具体见3)
names为自己训练的分类名称(具体见4)
2.VOC格式xml文件转换为YOLO格式的txt文件
使用脚本xmltotxt.py即可实现,主要功能有两个:
i:自动生成路径txt文件,三个txt文件中的包含了训练,测试和验证图片的具体路径。
ii:将xml文件转换为txt文件,YOLO需要识别txt文件
脚本文件代码:
#×××××××××××××××××××××××××××××××××××××××××××××××××××××××××#
#2018-10-9
#该脚本的主要功能:
#1