最近看了些yolov3和ssd的水文，感觉有些收获，所以基于自己的理解搭建了个人脸检测的模型，在winderface数据集上进行尝试性训练．有一点效果，但还有很大的训练和改进空间．这里把工程share出来，希望有相同爱好的可以一起优化，训练出一个state of art的人脸检测模型．源码地址(github)：face_detect如果对你有帮助，给他star鼓励下，谢谢！！！我的模型说明：基于resnet，Bottleneck是一个resnet单元．在三个特征map上检测人
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个人脸．
最低0.47元/天 解锁文章
4409
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
