失眠网 > Pytorch实现戴口罩人脸检测和戴口罩识别(含训练代码戴口罩人脸数据集)

Pytorch实现戴口罩人脸检测和戴口罩识别(含训练代码戴口罩人脸数据集)

时间：2023-11-19 05:47:43

1.戴口罩识别的方法

（1）基于多类别目标检测的戴口罩识别方法

（2）基于人脸检测+戴口罩分类识别方法

2.戴口罩人脸数据集

3.戴口罩人脸检测

4.戴口罩识别模型训练

（1）准备数据

（2）戴口罩分类模型训练(Pytorch)

（3）可视化训练过程

（4）戴口罩识别效果

（5）一些运行错误处理方法

5.项目源码下载（Python版）

6.项目源码下载（Android版）

当前疫情反反复复,而防控新冠病毒的最有效手段之一就是戴口罩,因此研究戴口罩检测和识别具有重大意义。疫情防控，人人有责，作为一名程序狗，分享一下鄙人开发的戴口罩人脸检测和戴口罩识别方法。项目开发基于深度学习框架Pytorch开发一套的戴口罩识别（face-mask recognition）识别系统，目前准确率还挺高的，在resnet50,可以高达99%的准确率，即使采用轻量化版本MobileNet-v2，准确率也可以高达98.18%左右。

【尊重原创，转载请注明出处】/guyuealian/article/details/125428609

先展示一下，Python版本的戴口罩人脸检测和戴口罩识别Demo效果：

戴口罩人脸检测和戴口罩识别整套Python代码下载：Pytorch实现戴口罩人脸检测和戴口罩识别(含训练代码戴口罩人脸数据集)

整套项目项，支持的主要内容主要有：

提供5个戴口罩人脸数据集： facemask-train1， facemask-train2，facemask-train3， synthetic-train1，synthetic-train2 ，facemask-test ，总共约有50000+的数据：提供生成戴口罩人脸代码： python create_facemask.py提供戴口罩分类识别训练代码：train.py提供戴口罩分类识别测试代码：demo.pydemo支持戴口罩人脸检测，支持戴口罩识别：mask(戴口罩)和nomask(未佩戴口罩)提供戴口罩识别Python Demo源码，在普通电脑CPU/GPU上可以实时检测和识别

如果你需要Android版本的戴口罩人脸检测和戴口罩识别，请参考《Android实现戴口罩人脸检测和戴口罩识别(附Android源码)》https://panjinquan./article/details/128404379

1.戴口罩识别的方法

（1）基于多类别目标检测的戴口罩识别方法

基于多类别目标检测的戴口罩识别方法，一步到位，把未戴口罩（nomask）和戴口罩（mask）两个类别直接当成两个目标检测的类别进行训练

优点：直接端到端训练，任务简单，速度快缺点：需要人工标注人脸框mask和nomask，时间花费比较大；训练数据不足的情况下，容易出现误检测的情况

（2）基于人脸检测+戴口罩分类识别方法

该方法，先采用通用的人脸检测模型，进行人脸检测，然后裁剪人脸区域，再训练一个戴口罩分类器，对人脸进行分类识别（未戴口罩和戴口罩）

优点：不需要标注人脸框数据，可以自己合成戴口罩人脸数据，人工成本低；精度高，可针对分类模型进行轻量化缺点：需要部署两个模型（人脸检测模型和戴口罩分类模型），人脸越多，速度越慢

考虑到数据标注成本的问题，本项目采用第二种方法，即采用基于人脸检测+戴口罩分类识别方法

2.戴口罩人脸数据集

网上绝大部分人脸数据都是不戴口罩的人脸，不能直接用于戴口罩识别中。鉴于此，我们可以考虑自己合成/生成戴口罩的人脸数据，以下是鄙人收藏和整理的戴口罩人脸数据集和合成的数据集，总共约有50000+的数据：

关于戴口罩人脸数据和生成方法，详细使用说明请参考我的一篇博客《戴口罩人脸数据集和戴口罩人脸生成方法》

3.戴口罩人脸检测

通常我们理解的人脸检测是指没有遮挡或者只有少许遮挡情况下的人脸检测，当人脸戴有口罩，其检测效果势必会变得比较差，而大量标注带有人脸口罩的人脸数据集还是比较耗时费力的。所以我的方法是：

先在WiderFace人脸数据集上，训练人脸检测；然后在facemask-train1数据集finetune人脸检测模型，经过这个方法训练后，其戴口罩检测效果会好很多。

当然，即使使用开源的人脸检测算法，在带有口罩人脸检测，其实效果也不会太差，比如使用FaceBox,MTCNN检测带有口罩的图片，效果也可以的，只不过会经常出现人脸检测框不完整，存在缺少等问题，对后续的戴口罩的识别有一定的影响。

关于人脸检测的方法，可以参考我的另一篇博客：《行人检测和人脸检测和人脸关键点检测（C++/Android源码）》

4.戴口罩识别模型训练

有了戴口罩人脸检测，接下来就可以开始训练戴口罩分类识别模型；考虑到后续我们需要将戴口罩识别模型部署到Android平台中，因此项目选择计算量比较小的轻量化模型（mobilenet_v2）

整套工程项目基本结构如下：

.├── classifier # 训练模型相关工具├── configs# 训练配置文件├── data # 训练数据├── libs│├── convert# 将模型转换为ONNX工具│├── facemask# 戴口罩人脸数据生成工具│├── light_detector # 人脸检测│├── create_facemask.py # 戴口罩人脸数据生成demo│├── detector.py # 人脸检测demo│└── README.md ├── demo.py # 戴口罩人脸识别demo├── README.md# 项目工程说明文档├── requirements.txt # 项目相关依赖包└── train.py# 训练文件

项目依赖python包请参考requirements.txt，使用pip安装即可：

numpy==1.16.3matplotlib==3.1.0Pillow==6.0.0easydict==1.9opencv-contrib-python==4.5.2.52opencv-python==4.5.1.48pandas==1.1.5PyYAML==5.3.1scikit-image==0.17.2scikit-learn==0.24.0scipy==1.5.4seaborn==0.11.2tensorboard==2.5.0tensorboardX==2.1torch==1.7.1+cu110torchvision==0.8.2+cu110tqdm==4.55.1xmltodict==0.12.0basetrainerpybaseutils==0.6.5

（1）准备数据

总共有5个数据集，包括facemask-train1， facemask-train2，facemask-train3，synthetic-train1，synthetic-train2 ，facemask-test ，总共约有50000+的数据。

当然，你也可以使用自己的数据集，数据结构如下，其中mask目录存放戴口罩的人脸图片，而nomask目录存放未戴口罩的人脸图像。

（2）戴口罩分类模型训练(Pytorch)

鄙人在《Pytorch基础训练库Pytorch-Base-Trainer(支持模型剪枝分布式训练)》基础上实现了戴口罩和未佩戴口罩二分类识别训练和测试，整套训练代码非常简单操作，用户只需要将相同类别的数据放在同一个目录下，并填写好对应的数据路径，即可开始训练了。

训练框架采用Pytorch，整套训练代码支持的内容主要有：

目前支持的backbone有：googlenet,resnet[18,34,50], ,mobilenet_v2等，其他backbone可以自定义添加训练参数可以通过(configs/config.yaml)配置文件进行设置

训练参数说明如下：

# 设置训练数据集，支持多个训练数据集train_data:- 'dataset/face_mask/facemask-train1/crops'- 'dataset/face_mask/facemask-train2/crops'- 'dataset/face_mask/facemask-train3/crops'- 'dataset/face_mask/synthetic-train1/crops'- 'dataset/face_mask/synthetic-train1/crops'# 设置测试数据集test_data: 'dataset/face_mask/facemask-test/crops'class_name: 'dataset/face_mask/class_name.txt' # 类别标签train_transform: "train" # 训练使用的数据增强方法test_transform: "val" # 测试使用的数据增强方法work_dir: "work_space/" # 保存输出模型的目录net_type: "mobilenet_v2" # 骨干网络,支持：resnet18,mobilenet_v2,googlenetresample: True # 进行样本均衡width_mult: 1.0input_size: [ 128,128 ]rgb_mean: [ 0.5, 0.5, 0.5 ] # for normalize inputs to [-1, 1],Sequence of means for each channel.rgb_std: [ 0.5, 0.5, 0.5 ] # for normalize,Sequence of standard deviations for each channel.batch_size: 64lr: 0.01 # 初始学习率optim_type: "SGD" # 选择优化器，SGD,Adamloss_type: "LabelSmoothing" # 选择损失函数：支持CrossEntropyLoss,LabelSmoothingmomentum: 0.9# SGD momentumnum_epochs: 100 # 训练循环次数num_warn_up: 3# warn-up次数num_workers: 8# 加载数据工作进程数weight_decay: 0.0005 # weight_decay，默认5e-4scheduler: "multi-step"# 学习率调整策略milestones: [ 20,50,80 ]# 下调学习率方式gpu_id: [ 0 ]# GPU IDlog_freq: 50 # LOG打印频率progress: True# 是否显示进度条pretrained: False # 是否使用pretrained模型finetune: False # 是否进行finetune

开始训练：

python train.py -c configs/config.yaml

训练完成后，训练集的Accuracy在99%以上，测试集的Accuracy在98%左右

（3）可视化训练过程

训练过程可视化工具是使用Tensorboard，使用方法：

# 基本方法tensorboard --logdir=path/to/log/# 例如tensorboard --logdir=work_space/mobilenet_v2_1.0_CrossEntropyLoss/log

可视化效果

（4）戴口罩识别效果

测试图片文件

# 测试图片image_dir='data/test_image' # 测试图片的目录model_file="data/pretrained/mobilenet_v2_1.0_CrossEntropyLoss/model/best_model_078_98.3498.pth" # 模型文件out_dir="output/" # 保存检测结果python demo.py --image_dir $image_dir --model_file $model_file --out_dir $out_dir

测试视频文件

# 测试视频文件video_file="data/video.mp4" # 测试视频文件，如*.mp4,*.avi等model_file="data/pretrained/mobilenet_v2_1.0_CrossEntropyLoss/model/best_model_078_98.3498.pth" # 模型文件out_dir="output/" # 保存检测结果python demo.py --video_file $video_file --model_file $model_file --out_dir $out_dir

测试摄像头

# 测试摄像头video_file=0 # 测试摄像头IDmodel_file="data/pretrained/mobilenet_v2_1.0_CrossEntropyLoss/model/best_model_078_98.3498.pth" # 模型文件out_dir="output/" # 保存检测结果python demo.py --video_file $video_file --model_file $model_file --out_dir $out_dir

（5）一些运行错误处理方法

cannot import name 'load_state_dict_from_url'

由于一些版本升级，会导致部分接口函数不能使用，请确保版本对应
torch==1.7.1
torchvision==0.8.2

或者将对应python文件将

from torchvision.models.resnet import model_urls, load_state_dict_from_url

修改为：

from torch.hub import load_state_dict_from_urlmodel_urls = {'mobilenet_v2': '/models/mobilenet_v2-b0353104.pth','resnet18': '/models/resnet18-5c106cde.pth','resnet34': '/models/resnet34-333f7ec4.pth','resnet50': '/models/resnet50-19c8e357.pth','resnet101': '/models/resnet101-5d3b4d8f.pth','resnet152': '/models/resnet152-b121ed2d.pth','resnext50_32x4d': '/models/resnext50_32x4d-7cdf4587.pth','resnext101_32x8d': '/models/resnext101_32x8d-8ba56ff5.pth','wide_resnet50_2': '/models/wide_resnet50_2-95faca4d.pth','wide_resnet101_2': '/models/wide_resnet101_2-32ee1156.pth',}

5.项目源码下载（Python版）

整套项目源码内容包含：Pytorch实现戴口罩人脸检测和戴口罩识别(含训练代码戴口罩人脸数据集)

提供5个戴口罩人脸数据集： facemask-train1， facemask-train2，facemask-train3， synthetic-train1，synthetic-train2 ，facemask-test ，总共约有50000+的数据：提供生成戴口罩人脸代码： python create_facemask.py提供戴口罩分类识别训练代码：train.py提供戴口罩分类识别测试代码：demo.pydemo支持戴口罩人脸检测，支持戴口罩识别：mask(戴口罩)和nomask(未佩戴口罩)提供戴口罩识别Python Demo源码，在普通电脑CPU/GPU上可以实时检测和识别