基于 Tensorflow 的人脸口罩识别系统的设计与实现代码大全.pdf

基于Tensorflow的人脸口罩识别系统的设计与实现代码大全--第1页

基于Tensorflow的人脸口罩识别系统的设计与实现

1.1题目的主要研究内容

（1）工作的主要描述

人脸口罩识别主要是可以有效检测在密集人流区域中携带和未携戴口罩的

所有人脸，同时判断该者是否佩戴口罩。本系统由两个功能单元组成，可以分别

完成口罩人脸的检测和口罩人脸的分类。

（2）系统流程图

批量图片数据

基础卷积层

辅助卷积层ssd网络目标定位和检测

预测卷积层

模型参数更

损失函数计算

新

1.2题目研究的工作基础或实验条件

（1）硬件环境

芯片：AppleM1

内存：8GB

基于Tensorflow的人脸口罩识别系统的设计与实现代码大全--第1页

基于Tensorflow的人脸口罩识别系统的设计与实现代码大全--第2页

（2）软件环境

开发语言：python（版本号python3.7）

开发软件：PyCharm

1.3数据集描述

本设计所用数据集是在公开数据集上找了将近8000张的人脸口罩数据和模

型，主要来源于WIDERface和MAFA数据集。其中包括了4064张佩戴口罩的

人脸图片，3894张未佩戴口罩的人脸图片。并将这些数据集分为测试集和验证

集两部分。

带口罩的数据集：

不戴口罩的数据集：

基于Tensorflow的人脸口罩识别系统的设计与实现代码大全--第2页

基于Tensorflow的人脸口罩识别系统的设计与实现代码大全--第3页

1.4特征提取过程描述

本设计采用了ssd架构的人脸检测算法，在这基础上增加一个类别分类，即

戴口罩和不戴口罩两类。在这个模型中输入是260*260大小的图片，主干网络只

有8个卷积层，有五个定位和分类层，一共具有28层，且每个卷积层的通道数

目基本都是32、64、128这三种。其网络结构为：

Conv+ReLu—Pooling—Conv+ReLu—Pooling—Conv+ReLu—Pooling—Conv

+ReLu—Pooling—Conv+ReLu—Pooling—Conv+ReLu—Pooling—Conv+ReLu—

Conv+ReLu—Conv—ReLu—Conv—Permute—Reshape—sigmoid，

其中前面八个卷积层为主干网络，主要应用于特征提取，最后面为定位和分

类层。在ssd的网络中原始图像经过卷积层转换后得到原图像的特征映射图，再

用多个卷积层对特征映射图处理后用来定位和检测原始图像。ssd的网络结构中

包含了基础网络，辅助卷积层和预测卷积层。其中基础网络用来提取低尺度的特

征映射图，辅助卷积层用来提取高尺度的特征映射图。

基础网络的结构采用了vgg-16的网络架构，因为VCG-16网络包含了卷积

层和全连接层（FCLayers），全连接层的任务用来分类，由于基础网络只需要

提取特征映射图，因此需要对全连接层用卷积层代替。辅助卷积层连接基础网络

最后的特征映射图，通过卷积神经网络输出4个高尺度的特征映射图