1. 程式人生 > >YOLO v2 人臉檢測——加入Wider_face資料集進行fine tuning[by zhangzexuan]

YOLO v2 人臉檢測——加入Wider_face資料集進行fine tuning[by zhangzexuan]

一、資料準備

Wider_face官方網站下載Wider_face資料集以及標籤,按照【http://www.cnblogs.com/linyuanzhou/p/6043436.html】提供的wider_face轉VOC資料集python程式進行資料格式轉換。


二、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

三、結果: