从LFW Face Database(http://vis-www.cs.umass.edu/lfw/#download) 中下载人脸图片数据,下面以一个人脸识别小数据集为例。
数据集使用LabelImg工具手动标注80张作为Training Set,20张作为Validation Set,保存相应xml文件。
将原图片(RawData)存放在JPEGImages,xml文件存放在Annotations。
在训练前,还需要在configuration中修改数据集路径为自己构建的小数据集。-
由于标注时label为’person’,PASCAL_VOC_CLASSES中尚没有对应字典项,-
需在PASCAL_VOC_CLASSES中添加’person’,类别与’face’相同,均为1,(或直接使用voc.names文件)然后运行write_voc_to_txt.py,将voc转换为txt内的向量形式,即可开始训练。