在Pascal VOC 数据集上训练YOLOv3模型

     上节介绍了《从零开始在Windows10中编译安装YOLOv3》,本节介绍在Pascal VOC 数据集上训练YOLOv3。

第一步,下载并安装YOLOv3训练依赖项。

a.下载Pascal VOC 数据集

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

到darknet\build\darknet\x64\data\voc文件夹,然后分别右键点击三个压缩文件,在右键菜单中选择“解压到当前文件夹(X)”,这样可以获得一个VOCdevkit文件夹,里面包含VOC数据集。

下载并解压VOC数据集

b.下载安装Anaconda,下载链接:https://www.anaconda.com/distribution/,注意安装的时候,勾选: “Add

Anaconda to my PATH environment variable”选项

安装Anaconda

c.运行voc_label.py。voc_label.py应该在darknet\build\darknet\x64\data\voc文件夹下,若没有,请到http://pjreddie.com/media/files/voc_label.py处下载,并放到voc文件夹下。

在darknet\build\darknet\x64\data\voc文件夹的地址栏中键入“cmd”,启动Windows命令行终端,运行命令:

python voc_label.py

这样可以获得2007_test.txt,2007_train.txt,2007_val.txt,2012_train.txt,2012_val.txt五个文件,运行命令:

type 2007_train.txt 2007_val.txt 2012_*.txt > train.txt

获得train.txt文件

第二步,修改yolov3-voc.cfg文件。在darknet\build\darknet\x64\cfg文件夹下,用文本编辑器打开yolov3-voc.cfg文件,修改第3~4行如下图所示。

batch=64,意思是每batch个样本更新一次参数。

subdivisions=16,意思是如果内存不够大,将batch分割为subdivisions个子batch,每个子batch的大小为batch/subdivisions。如果出现“Out of memory”  错误,则把 subdivisions依次增大:16, 32 or 64,直到不发生错误为止。

修改yolov3-voc.cfg

第三步,修改voc.data文件。在darknet\build\darknet\x64\data文件夹中,用文本编辑器打开voc.data文件,修改第2~6行如下图所示

修改voc.data文件

最后一步,在darknet\build\darknet\x64文件夹的地址栏中键入“cmd”,启动Windows命令行终端,运行命令:

d:\darknet\darknet.exe detector train cfg\voc.data cfg\yolov3-voc.cfg darknet53.conv.74

启动训练,如下图所示

训练YOLOv3

训练完毕后,在darknet\build\darknet\x64\backup文件夹可以看到每隔1000次迭代得到的权重文件:

权重文件
训练结果

在D:\darknet\build\darknet\x64> 下用命令:

d:\darknet\darknet.exe detector test cfg\voc.data cfg\yolov3-voc.cfg backup\yolov3-voc_final.weights data\voc\VOCdevkit\VOC2007\JPEGImages\000004.jpg

测试训练的权重文件

参考文献:

[1] https://github.com/AlexeyAB/darknet

[2] https://pjreddie.com/darknet/yolo/

[3]《在Windows下Darknet.exe的命令参数

下节介绍

在COCO 数据集上训练YOLOv3模型

在自己的数据集上训练YOLOv3模型

参考阅读《深度学习图像识别技术

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