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的大小,表示到目前为止,参与训练的图片的总量。

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 213,014评论 6 492
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,796评论 3 386
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 158,484评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,830评论 1 285
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,946评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,114评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,182评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,927评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,369评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,678评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,832评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,533评论 4 335
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,166评论 3 317
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,885评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,128评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,659评论 2 362
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,738评论 2 351

推荐阅读更多精彩内容