foamliu / YOLO-Face-Detection

YOLO 人脸检测

基于YOLOv2的人脸检测之Keras 实现。

依赖项

• NumPy
• Tensorflow
• Keras
• OpenCV

数据集

WIDER FACE 数据集，32,203个图像并标记393,703张人脸；随机选择40％/ 10％/ 50％的数据作为训练，验证和测试集。

请按照说明 下载 WIDER_train.zip, WIDER_val.zip, WIDER_test.zip 以及 wider_face_split.zip 放入 data 目录。

$ wget http://mmlab.ie.cuhk.edu.hk/projects/WIDERFace/support/bbx_annotation/wider_face_split.zip

用法

数据预处理

提取12880张训练图片，3226张验证图片和16097张测试图片：

$ python pre-process.py

训练

$ python train.py

如果想在培训期间进行可视化，请在终端中运行：

$ tensorboard --logdir path_to_current_dir/logs

Demo

下载 pre-trained model 放在 models 目录然后执行:

$ python demo.py

1	2	3	4

数据增强

$ python augmentor.py

before	after