集大成！10 种轻量级人脸检测算法的对比 ( 含源码 )
（给Python开发者加星标，提升Python技能）
来源：github-hpc203
作者实现了10种常用的轻量级人脸检测算法，本套程序极简，高效，尽可能少的依赖第三方库，是本套程序的特点。
安装
在下载完程序后，需要下载模型文件，才能正常运行程序。由于模型文件太多，无法直接上传，可以去百度云盘下载：
链接: https://pan.baidu.com/s/16EBjFG5tj6rp0m8TadDgHg 提取码: rqfb
下载完成后一共有12个文件夹，把它们放到本仓库代码文件夹里。程序依赖pytorch和opencv，如果你的python环境里没有这两个库，那在Terminal输入pip安装。
pytorch的安装命令是 pip install torch 和 pip install torchvision 而opencv的安装命令是 pip install opencv-python 配置好运行环境之后，就可以运行程序了。
使用方法
运行 Run_all_compare_time.py，就可以看到10种人脸检测算法的结果和运行耗时统计直方图。
每一种人脸检测算法，作者都用一个类包装起来的，在这个类有构造函数__init__和成员函数detect。其中 centerface，dbface，retinaface，mtcnn，yoloface，libface 这六个人脸检测网络在输出检测框的同时还输出人脸里的5个关键点， 用这5个关键点可以做人脸对齐的。它们的类构造函数里有个初始化参数align是用来决定是否做人脸对齐的开关，在人脸识别系统里，人脸对齐这一步不是必选项的。
如果你想构建一个人脸识别系统，那可以先运行get_face_feature.py，它是获取人脸特征向量的，最后会生成一个已知身份的人脸特征向量的pkl文件。在这个.py文件里，使用的人脸检测器是yoloface，如果你想换用其它的人脸检测器，那就修改主函数开头的from...import...即可。假如你使用的人脸检测器的输出没有关键点，而你想在人脸检测后做人脸对齐，那么检测人脸关键点的功能可以使用 mtcnn_pfld_landmark.py里的类pfld_landmark，它就是PFLD人脸关键点检测，需要注意的是，它的输入是在人脸检测之后剪切出的人脸roi区域。接下来是提取人脸特征向量，用的是arcface。
在得到人脸特征向量的pkl文件后，运行detect_face_align_rec.py，就是做人脸检测→人脸对齐→人脸识别，需要注意的是 在上一步提取已知身份的人脸特征向量的pkl文件时，get_face_feature.py使用的人脸检测器和现在detect_face_align_rec.py里使用的人脸检测器一致。
看近几年顶会的paper，在人脸检测→人脸对齐→人脸识别这三大模块里，做的最多的是人脸检测这个模块，几乎每年都有新的网络提出，作者关心和感兴趣的也是人脸检测这个模块。而在人脸识别提取特征向量这个模块，创新性的工作都是在最后的全连接层后面的loss做改动创新，怎样让同类人脸更加聚拢，让不同类人脸的间距更大。在提取人脸特征向量这一步，目前也有很多种网络，比如ArcFace、SphereFace、CosFace等等的，如果想继续扩充现在的程序，读者可以把这些模块添加进来。
需要注意的是，这里面有blazeface，作者是参考
https://github.com/hollance/BlazeFace-PyTorch/blob/master/Inference.ipynb
把blazeface修改成类的形式，程序能正常运行的。但是blazeface要求输入图片是128x128的，这个分辨率太低了， 用本代码文件夹里的两幅图片resize到128x128，给blazeface做人脸检测的效果太差。作者没将该算法加入到Run_all_compare_time.py里做比较的。
拓展
经过这几天运行程序比较，作者想要做一个实时人脸识别系统，libfacedetect是做人脸检测的最佳选择， 此外，作者把PFLD人脸关键点检测模型转换成onnx格式的，这样就可以使用opencv的dnn模块做PFLD人脸关键点检测。
这部分代码在libfacedetect_align_module.py里。作者编写了类将libfacedetect和PFLD串联起来， 仅仅使用opencv的dnn模块，实现一个人脸+关键点检测的功能。
另外，作者编写了retinaface的dnn版本，使得用retinaface做人脸检测，只依赖opencv的dnn模块，解码和nms后处理依赖numpy， 从而使retinaface的整个人脸检测流程彻底摆脱对pytorch的依赖。
GitHub地址：
https://github.com/hpc203/10kinds-light-face-detector-align-recognition
- EOF -
觉得本文对你有帮助？请分享给更多人
推荐关注「Python开发者」，提升Python技能
点赞和在看就是最大的支持❤️