YOLO v2 人脸检测——加入Wider_face数据集进行fine tuning[by zhangzexuan]


一、数据准备

Wider_face官方网站下载Wider_face数据集以及标签,按照【http://www.cnblogs.com/linyuanzhou/p/6043436.html】提供的wider_face转VOC数据集python程序进行数据格式转换。

使用该python程序时的目录结构如下所示(此图摘自http://www.cnblogs.com/linyuanzhou/p/6043436.html):

YOLO v2 人脸检测——加入Wider_face数据集进行fine tuning[by zhangzexuan]

二、Fine tuning

按照上一篇博文的流程,合并一下CelebA和Wider_face转换成YOLO格式的数据集以及labels,根据需要在darknet/cfg/tiny-yolo.cfg中进行参数调整即可。

使用已经训练好的权重文件继续训练的命令如下:

./darknet detector train cfg/voc.datacfg/tiny-yolo.cfg [权重文件] -gpus0,1,2,3

三、结果:

YOLO v2 人脸检测——加入Wider_face数据集进行fine tuning[by zhangzexuan]