YbFase(yolo base face detector),是一个借鉴了yolov3的网络结构,在多特征尺度下预测人脸框的人脸检测算法.并结合了yolov4使用的mish激活函数.该模型大小仅为7,8M,无论模型大小,还是准确率,还是执行速率都很有竞争力.
源码地址(github):
如果对你有帮助,给他star鼓励下,谢谢!!!
完全开源
很多开源的人脸检测项目,或者没有训练代码,或者没有评估代码,而我的这个YbFace项目,包含了做人脸检测你需要的所有东西.
- 训练代码
- 测试代码
- 计算ap的评估代码及其可视化
- 模型可视化
模型
tensorboard绘制的模型结构图:
我绘制的模型结构图:
说明:
- 基于resnet,Bottleneck是一个resnet单元.
- 在三个特征map上检测人脸,分别是13*13, 26*26, 52*52,因此,每张图片最多能检测13*13+ 26*26+ 52*52个人脸.