基于YOLOv2的人脸检测之Keras 实现。
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
下载 pre-trained model 放在 models 目录然后执行:
$ python demo.py
1 | 2 | 3 | 4 |
---|---|---|---|
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
$ python augmentor.py
before | after |
---|---|
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |
![]() |