Windows7 搭档yolov3 重新训练voc数据

官网: http://pjreddie.com/yolo/ 

相关文章: http://arxiv.org/abs/1506.02640 

源代码: https://github.com/pjreddie/darknet.git

1、数据集制作

A.制作VOC格式的xml文件

工具:LabelImg  GitHub - tzutalin/labelImg: LabelImg is a graphical image annotation tool and label object bounding boxes in images

B.将VOC格式的xml文件转换成YOLO格式的txt文件

因为我这里使用的是VOC数据集,不过减少分类了,所以直接用已有的数据集

首先下载预训练的权重文件: http://pjreddie.com/media/files/darknet53.conv.74 放在build\darknet\x64文件夹下

下载Pascal VOC 数据集并且放在 build\darknet\x64\data\voc路径下,然后创建路径 build\darknet\x64\data\voc\VOCdevkit\:

http://pjreddie.com/media/files/VOCtrainval_11-May-2012.tar

http://pjreddie.com/media/files/VOCtrainval_06-Nov-2007.tar

http://pjreddie.com/media/files/VOCtest_06-Nov-2007.tar

下载 voc_label.py 并放在 build\darknet\x64\data\voc: http://pjreddie.com/media/files/voc_label.py

安装python,建议安装3.5以上的版本

在cmd命令行下运行以下指令: python build\darknet\x64\data\voc\voc_label.py (to generate files: 2007_test.txt, 2007_train.txt, 2007_val.txt, 2012_train.txt, 2012_val.txt)

生成对应的txt文件,在这里,一直报我的test生成路径找不到,因为暂时没用,我没做处理

接着运行指令: type 2007_train.txt 2007_val.txt 2012_*.txt > train.txt

生成训练数据的label文件。

2、文件修改:

(A)关于 .data .names 两个文件修改非常简单,打开cfg/voc.data文件,进行如下修改:

classes= 5 # 自己数据集的类别数

train  = E:\\Codes\\DLdetection\\darknet\\train.txt

valid  = E:\\Codes\\DLdetection\\darknet\\2007_test.txt

names = E:\\Codes\\DLdetection\\darknet\\data\\voc.names

backup = E:\\Codes\\DLdetection\\darknet

注:需要在darknet文件夹下,新建名为backup的文件夹,否则训练过程报错:Couldn't open file: backup/yolov3-voc.backup。

打开data/voc.names文件,对应自己的数据集修改类别。

转换标注集格式

修改voc_label.py, 如只有一个类:car,在voc_label.py中做如下修改

sets=[('2007','train'), ('2007','val'), ('2007','test')]

classes = ["car"]'''

#classes = ["aeroplane", "bicycle", "bird", "boat", "bottle", "bus", "car", "cat", "chair", "cow", "diningtable", "dog", "horse", "motorbike", "person", "pottedplant", "sheep", "sofa", "train", "tvmonitor"]

运行文件voc_label.py

python E:\Codes\DLdetection\darknet\build\darknet\x64\data\voc\voc_label.py

会在目录下生成一系列文件并将VOC格式标注转为YOLO格式txt标注(归一化处理)

(B)关于cfg修改,以5类目标检测为例,查找带有[convolutional]以及[yolo]标签处(共3处)。#表示注释,根据训练和测试,自行修改。

[net]

# Testing

# batch=1

# subdivisions=1

# Training

 batch=64

 subdivisions=8

......

[convolutional]

size=1

stride=1

pad=1

filters=33###75


activation=linear

[yolo]

mask = 6,7,8

anchors = 10,13,  16,30,  33,23,  30,61,  62,45,  59,119,  116,90,  156,198,  373,326

classes=5###20

num=9

jitter=.3

ignore_thresh = .5

truth_thresh = 1

random=0###1


......

[convolutional]

size=1

stride=1

pad=1

filters=33###75

activation=linear

[yolo]

mask = 3,4,5

anchors = 10,13,  16,30,  33,23,  30,61,  62,45,  59,119,  116,90,  156,198,  373,326

classes=5###20

num=9

jitter=.3

ignore_thresh = .5

truth_thresh = 1

random=0###1

......

[convolutional]

size=1

stride=1

pad=1

filters=33###75

activation=linear

[yolo]

mask = 0,1,2

anchors = 10,13,  16,30,  33,23,  30,61,  62,45,  59,119,  116,90,  156,198,  373,326

classes=5###20

num=9

jitter=.3

ignore_thresh = .5

truth_thresh = 1

random=0###1

A.filters数目是怎么计算的:3x(classes数目+1+4),和聚类数目分布有关,论文中有说明;

B.如果想修改默认anchors数值,使用k-means即可;

C.如果显存很小,将random设置为0,即关闭多尺度训练;

D.其他参数如何调整,有空再补;

E.前100次迭代loss较大,后面会很快收敛;

3、训练,以上步骤结束后,在cmd下运行以下指令:darknet.exe detector train data/voc.data cfg/yolov3-voc.cfg darknet53.conv.74 开始进行训练


训练截图

一次迭代

主要参数说明:

Region xx: cfg文件中yolo-layer的索引;

Avg IOU:当前迭代中,预测的box与标注的box的平均交并比,越大越好,期望数值为1;

Class: 标注物体的分类准确率,越大越好,期望数值为1;

obj: 越大越好,期望数值为1;

No obj: 越小越好;

.5R: 以IOU=0.5为阈值时候的recall; recall = 检出的正样本/实际的正样本

0.75R: 以IOU=0.75为阈值时候的recall;

count:正样本数目。

9798: 指示当前训练的迭代次数

0.370096: 是总体的Loss(损失)

0.451929 avg: 是平均Loss,这个数值应该越低越好,一般来说,一旦这个数值低于0.060730 avg就可以终止训练了。

0.001000 rate: 代表当前的学习率,是在.cfg文件中定义的。

3.300000 seconds: 表示当前批次训练花费的总时间。

627072 images: 这一行最后的这个数值是9798*64的大小,表示到目前为止,参与训练的图片的总量。

你可能感兴趣的:(Windows7 搭档yolov3 重新训练voc数据)