软硬件环境
-
windows 10 64bit -
DBFace -
pytorch 1.6 -
opencv 4.4.0
简介
DBFace是一个开源的轻量级(模型只有7兆)的实时人脸检测方法，检测的速度与精度都很是不错。下图展现了多种人脸检测方法在WiderFace数据集上的测试效果，能够看到不只DBFace模型的大小最小，并且在Easy、medium、Hard三个测试任务中均取得了最高的检测精度。html
下图展现了不一样人脸检测方法在WiderFace数据集上的P-R曲线，能够看出，DBFace包围的面积在三个任务中均相对较大。python
安装测试
首先建立python虚拟环境git
conda create -n dbface python=3.7
conda activate dbface
接下来克隆DBFace的源码并安装必要的依赖库github
git clone https://github.com/dlunion/DBFace.git
cd DBFace
# 安装opencv
pip install opencv-python
# 安装pytorch1.6和cuda10.1
pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html
修改main.py文件第54行，由web
ys = list((indices / hm_width).int().data.numpy())
改成windows
ys = list((torch.true_divide(indices, hm_width)).int().data.numpy())
而后执行测试脚本微信
python main.py
程序读取datas下的图片文件进行人脸检测，而后将检测结果保存在detect_result文件夹下编辑器
能够看到在这张全世界最大的自拍图片中，DBFace表现的很是棒，不少人脸，就算是人的肉眼，也很难分辨出。ide
除了检测图片中的人脸，测试代码中还提供了对usb camera的支持测试
参考资料
-
https://github.com/dlunion/DBFace -
https://xugaoxiang.com/2019/12/08/anaconda/
本文分享自微信公众号 - 迷途小书童的Note（Dev_Club）。
若有侵权，请联系 support@oschina.cn 删除。
本文参与“OSC源创计划”，欢迎正在阅读的你也加入，一块儿分享。