CN112949572B - 基于Slim-YOLOv3的口罩佩戴情况检测方法 - Google Patents
基于Slim-YOLOv3的口罩佩戴情况检测方法 Download PDFInfo
- Publication number
- CN112949572B CN112949572B CN202110330611.2A CN202110330611A CN112949572B CN 112949572 B CN112949572 B CN 112949572B CN 202110330611 A CN202110330611 A CN 202110330611A CN 112949572 B CN112949572 B CN 112949572B
- Authority
- CN
- China
- Prior art keywords
- convolution
- network
- mask
- prediction
- slim
- Prior art date
- Legal status (The legal status is an assumption and is not a legal conclusion. Google has not performed a legal analysis and makes no representation as to the accuracy of the status listed.)
- Active
Links
Images
Classifications
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V40/00—Recognition of biometric, human-related or animal-related patterns in image or video data
- G06V40/10—Human or animal bodies, e.g. vehicle occupants or pedestrians; Body parts, e.g. hands
- G06V40/16—Human faces, e.g. facial parts, sketches or expressions
- G06V40/161—Detection; Localisation; Normalisation
- G06V40/165—Detection; Localisation; Normalisation using facial parts and geometric relationships
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/045—Combinations of networks
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/04—Architecture, e.g. interconnection topology
- G06N3/048—Activation functions
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06N—COMPUTING ARRANGEMENTS BASED ON SPECIFIC COMPUTATIONAL MODELS
- G06N3/00—Computing arrangements based on biological models
- G06N3/02—Neural networks
- G06N3/08—Learning methods
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V40/00—Recognition of biometric, human-related or animal-related patterns in image or video data
- G06V40/10—Human or animal bodies, e.g. vehicle occupants or pedestrians; Body parts, e.g. hands
- G06V40/16—Human faces, e.g. facial parts, sketches or expressions
- G06V40/168—Feature extraction; Face representation
- G06V40/171—Local features and components; Facial parts ; Occluding parts, e.g. glasses; Geometrical relationships
-
- G—PHYSICS
- G06—COMPUTING; CALCULATING OR COUNTING
- G06V—IMAGE OR VIDEO RECOGNITION OR UNDERSTANDING
- G06V40/00—Recognition of biometric, human-related or animal-related patterns in image or video data
- G06V40/10—Human or animal bodies, e.g. vehicle occupants or pedestrians; Body parts, e.g. hands
- G06V40/16—Human faces, e.g. facial parts, sketches or expressions
- G06V40/172—Classification, e.g. identification
Landscapes
- Engineering & Computer Science (AREA)
- Physics & Mathematics (AREA)
- Health & Medical Sciences (AREA)
- Theoretical Computer Science (AREA)
- General Health & Medical Sciences (AREA)
- Oral & Maxillofacial Surgery (AREA)
- General Physics & Mathematics (AREA)
- Biophysics (AREA)
- Software Systems (AREA)
- Evolutionary Computation (AREA)
- Computational Linguistics (AREA)
- Molecular Biology (AREA)
- Computing Systems (AREA)
- General Engineering & Computer Science (AREA)
- Biomedical Technology (AREA)
- Mathematical Physics (AREA)
- Data Mining & Analysis (AREA)
- Artificial Intelligence (AREA)
- Life Sciences & Earth Sciences (AREA)
- Human Computer Interaction (AREA)
- Multimedia (AREA)
- Geometry (AREA)
- Computer Vision & Pattern Recognition (AREA)
- Image Analysis (AREA)
Abstract
本发明属于深度学习目标检测和计算机视觉技术领域,具体涉及一种基于Slim‑YOLOv3的口罩佩戴情况检测方法,该方法包括:实时获取人脸视频数据,对人脸视频数据进行预处理;将预处理后的人脸图像输入到训练好的Slim‑YOLOv3模型中,判断该用户是否正确佩戴口罩;本发明通过一种基于Slim‑YOLOv3的口罩佩戴情况视频检测方法,加上采用改进的无监督自分类方法对不规范佩戴口罩的数据进行子类划分,使得口罩佩戴视频检测任务可以更加精确快速的实现。且提出的网络更加简洁,使得应用成本进一步降低。
Description
技术领域
本发明属于深度学习目标检测和计算机视觉技术领域,具体涉及一种基于Slim-YOLOv3的口罩佩戴情况检测方法。
背景技术
由于有害的气体、气味、飞沫、病毒等均是通过空气侵入人体,通过规范佩戴口罩可以有效的防范该物质侵入人体。规范佩戴口罩的作用不仅仅是防止病毒从无症状者身上传播给他人,降低二次传播的概率以保护其他人,同时也可以保护佩戴者,减少佩戴者所接触的病毒的接种量,使得病毒感染风险更低。
近年来,深度学习在目标检测、图像分类、语义分割等领域取得了巨大的进展。结合卷积神经网络的各类算法,无论是精度还是运算速度都取得了巨大的进展。口罩佩戴情况视频检测任务是一个目标检测问题,目标检测是结合目标分类和目标定位的多任务深度学习问题。
目前,根据实际检测任务的需求,视频检测需要解决两大关键技术:
(1).实时性;现场视频检测需要保证实时,才能有效的捕捉当前任务对象口罩佩戴情况;
(2).高精度;只有精确的得出当前对象口罩佩戴情况,才能起到有效的辅助作用。
目前,虽然在实际应用中已经出现很多口罩佩戴情况视频检测的设备,但往往高检测精度的设备消耗的计算资源较贵,廉价的检测器无法达到高精度。
发明内容
为解决以上现有技术存在的问题,本发明提出了一种基于Slim-YOLOv3的口罩佩戴情况检测方法,该方法包括:实时获取人脸视频数据,对人脸视频数据进行预处理;将预处理后的人脸图像输入到训练好的改进Slim-YOLOv3模型中,判断该用户是否正确佩戴口罩;所述改进的Slim-YOLOv3模型包括骨干网络Darknet-53、特征增强及预测网络以及解码网络;
训练改进的Slim-YOLOv3模型的过程包括:
S1:获取原始数据集,对原始数据集进行预处理,得到训练样本集和测试样本集;
S2:对训练样本集和测试样本集中的数据进行分类和重新标注处理;
S3:将分类后的训练样本集输入到骨干网络Darknet-53中进行多尺度变换,提取多个尺度特征;
S4:将多个尺度特征输入到特征增强及预测网络中,得到分类预测结果;
S5:将分类预测结果输入到解码网络中进行解码;
S6:根据解码结果计算模型的损失函数;
S7:将测试集中的数据输入到模型进行预测,根据预测结果对模型的损失函数进行优化,当损失函数变化较小或达到迭代次数时,完成模型的训练。
优选的,对原始数据集进行预处理包括:将原始数据集中的数据进行压缩、翻转以及改变图像的明暗程度,得到增强的图像数据;对增强后的图像数据进行分割,得到训练样本集和测试样本集。
优选的,对训练样本集和测试样本集中的数据进行分类的过程包括:根据原始数据集的图像人脸佩戴口罩情况划分为三类,包括规范佩戴口罩图、不规范佩戴口罩图以及不佩戴口罩图;采用改进的图像无监督自分类方法SCAN对不规范佩戴口罩图进行再次分类,得到了多个子类。
进一步的,采用改进的图像无监督自分类方法SCAN对不规范佩戴口罩图进行再次分类的过程包括:
步骤1:提取口罩数据集中不规范佩戴口罩的人脸区域作为训练集;
步骤2:采用ECAResnet50网络对口罩佩戴情况数据集人脸区域数据进行分类训练,得到预训练权重;
步骤3:将预训练权重导入ECAResnet50的对抗网络中,提取图像的高级语义特征;
步骤4:计算各个高级语义特征的cosine相似度,并将相似度较大的语义特征对应的图像划分为邻居;
步骤5:将最近邻居作为先验进行聚类学习;
步骤6:采用self-labeling标签对聚类学习的图像进行微调标记处理,得到四个类别的伪标签。
进一步的,计算高级语义特征的cosine相似度的公式为:
其中,xi和yi分别表示两个语义特征的向量中的第i维,n表示向量的总维数。
优选的,采用骨干网络ECADarknet-53提取分类后的训练样本集中图像的多尺度特征的过程包括:将图像输入到数据增强模块中,将图像调整为416*416*3大小;将调整后的图像输入到ECADarknet53网络中,采用一个卷积快对图像进行一次卷积升维,得到一个batch_size*416*416*32大小的图像;采用引入注意力机制ECANet模块的五个残差卷积块对卷积升维后的图形进行特征提取,每经过一个残差卷积块,所提取的特征尺度变大,最终输出第四残差卷积块和第五残差卷积块得到的两个特征层;其中,batch_size表示每次输入网络的图像数量。
进一步的,注意力机制ECANet模块对特征进行处理的过程包括:采用不降低维数的通道级对特征层进行全局平均池化操作;每个通道选取k个近邻通道的数据对全局平均池化操作后的数据进行1×1的卷积,并通过一个sigmod激活函数;将激活后的数据扩展到输入特征大小并与输入特征相乘,得到含有多个通道信息的增强特征。ECANet模块被添加到ECADarknet-53中五个残差卷积块中的每个残差卷积单元中,每个残差卷积单元由两次卷积跟一个ECA模块的输出和输入叠加得到卷积单元的输出。
优选的,采用特征增强及预测网络对多个尺度特征进行处理的过程包括:
步骤1:将ECADarknet53网络中的第五残差卷积块得到的特征进行五次卷积处理;
步骤2:将进行卷积处理后的特征再次进行一次3×3卷积和一次1×1的卷积处理,将该处理的结果作为第五残差卷积块对应尺度特征层的预测结果;
步骤3:对五次卷积处理后的特征进行反卷积UmSampling2d操作后与第四残差卷积块得到的特征层进行堆叠处理,融合增强两个尺度特征的信息;
步骤4:对融合特征图进行五次卷积处理,将进行五次卷积的特征图进行一次3×3卷积和一次1×1的卷积,得到第四残差卷积后块对应尺度特征层的预测结果;
步骤5:输出两个尺度的特征层的预测结果,每个尺度的预测结果包括两个先验框每个网格点对应的预测框及其种类,即两个特征层分别对应着图片被分为不同大小的网格后,每个网格点上三个先验框对应的位置、置信度及其种类。
优选的,将分类预测结果输入到解码网络中进行解码的过程包括:
步骤1:将每个网格点加上对应的x_offset和y_offset,得到预测框的中心;
步骤2:将先验框和h、w结合,并计算出预测框的长和宽;
步骤3:通过位置信息与实际标注信息计算定位损失,通过预测类别信息和实际标注类别信息计算分类损失;
步骤5:判断真实框在图片中的位置,判断其属于哪一个网格点去检测;
步骤6:计算真实框与先验框重合程度,并选取重合度最高的先验框进行验证;
步骤7:获得网络应该有的预测结果,将其与实际的预测结果对比。
优选的,模型的损失函数的表达式为:
优选的,模型在对数据的种类进行分类过程中导入预训练权重,先冻结主干网后续网络层的参数进行50次迭代训练,然后解冻迭代训练100次,取其中分类损失和总损失较低的权重作为最终的训练结果。
本发明的有益效果:
本发明通过一种基于YOLOv3的口罩佩戴情况视频检测方法,使得口罩佩戴视频检测任务可以更加精确快速的实现。且提出的网络更加简洁,使得应用成本进一步降低。通过对数据集进一步进行子类划分和在主干网中添加ECANet注意力机制模块,提升了网络的检测精度;通过删除YOLOv3中对最小尺度的网络特征层,使得网络更加专注于大中尺度的目标,网络检测速度进一步提升。
附图说明
图1为本发明中数据集三个大类的划分示例图;
图2为本发明中不规范佩戴口罩的子类划分示例图;
图3为本发明中的原始YOLOv3的主要网络结构图;
图4为本发明中的ECANet的网络结构图;
图5为本发明提出的口罩佩戴视频检测任务的主要网络结构图;
图6为本发明的视频传输和显示设备图。
具体实施方式
下面将结合本发明实施例中的附图,对本发明实施例中的技术方案进行清楚、完整地描述,显然,所描述的实施例仅仅是本发明一部分实施例,而不是全部的实施例。基于本发明中的实施例,本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例,都属于本发明保护的范围。
一种基于Slim-YOLOv3的口罩佩戴情况检测方法,该方法包括:实时获取人脸视频数据,对人脸视频数据进行预处理;将预处理后的人脸图像输入到训练好的改进Slim-YOLOv3模型中,判断该用户是否正确佩戴口罩。所述改进的Slim-YOLOv3模型包括骨干网络Darknet-53、特征增强及预测网络以及解码网络。
训练改进的Slim-YOLOv3模型的过程包括:
S1:获取原始数据集,对原始数据集进行预处理,得到训练样本集和测试样本集;
S2:对训练样本集中的数据进行初始分类;
S3:将分类后的训练样本集中的图像输入到骨干网络Darknet-53中进行多尺度变换,提取多个尺度特征;
S4:将多个尺度特征输入到特征增强及预测网络中,得到分类预测结果;
S5:将分类预测结果输入到解码网络中进行解码;
S6:根据解码结果计算模型的损失函数;
S7:将测试集中的数据输入到模型进行测量,根据测量结果对模型的损失函数进行优化,当损失函数最小时,完成模型的训练。
一种训练改进的Slim-YOLOv3模型的具体实施方式,包括:
S1:获取原始数据集,并对原始数据集进行分类,分类的结果包括:规范佩戴口罩图、不规范佩戴口罩图以及不佩戴口罩图三类;
S2:将分类的数据集进行划分,得到训练样本集和测试样本集;对训练样本集进行数据增强处理;
S3:将增强后的训练样本集中的图像输入到骨干网络Darknet-53的YOLOv3网络模型中进行多尺度变换,提取多尺度分类特征和定位特征;
S4:需要输出两个特征层,两个特征层位于主干部分darknet53的不同位置,分别位于中下层,底层,两个特征层的大小分别为(26,26,512)和(13,13,1024),然后对两个特征层进行5次卷积处理;
S5:处理完后大小为13×13的特征层一部分用于输出该特征层对应的预测结果,一部分用于进行反卷积UmSampling2d后与大小为26×26的特征层进行结合,之后对两个尺度的特征图进行一次3×3和1×1的卷积处理;
S6:对于一张图片,若初始将其划分为K×K的网格,K表示截取的图片网格数量,越大表示截取的网格尺度越小;需要两个尺度,预测C个类别,那么最终每个规模得到的张量为K×K×[2×(4+1+C)]。其中4和1分别表示预测位置的x,y坐标偏移量x_offset和y_offset,预测目标的宽h和高w,以及目标的预测类别。网络使用多个独立的逻辑回归分类器进行分类,每个分类器对于目标边框中出现的物体只判断其是否属于当前标签,即简单的二分类,这样便实现了多标签分类。
S7:解码过程。通过将每个网格点加上它对应的x_offset和y_offset,加完后的结果就是预测框的中心,然后再利用先验框和h、w结合,计算出预测框的长和宽,然后通过位置信息与实际标注信息计算定位损失,通过预测类别信息和实际标注类别信息计算分类损失,过程如下:
1.判断真实框在图片中的位置,判断其属于哪一个网格点去检测;
2.判断真实框和哪个先验框重合程度最高;
3.计算该网格点应该有怎么样的预测结果才能获得真实框;
4.对所有真实框进行如上处理;
5.获得网络应该有的预测结果,将其与实际的预测结果对比。
S8:根据分类损失和定位损失,当损失收敛到一定程度,不在下降或者达到一定的迭代次数时,完成模型的训练。
如图1所示,将获取的数据根据口罩的实际佩戴情况划分为三个大类,即即不戴口罩(non-wearing mask)、不规范戴口罩(Wrong way wearing mask)和规范戴口罩(Standard wearing mask)。标签分别为Nomask,Wrmask和Swmask。
进一步的,不规范佩戴口罩存在较大的类间差距,会影响该类的检测精度,从而影响整体检测精度。将不规范戴口罩这一类数据集再重新分成四个子类,标签分别为Notnorm1、Notnorm2、Notnorm3和Notnorm4。
采用改进的图像无监督自分类方法SCAN对不规范佩戴口罩图进行分类的过程包括:
步骤1:提取口罩数据集中不规范佩戴口罩的人脸区域作为训练集;
步骤2:通过自监督方法提取图像的高级语义特征,消除当前的端到端学习方法中的低级特征;
步骤3:采用ECAResnet50网络对口罩佩戴情况数据集人脸区域数据进行分类训练,得到预训练权重;
步骤4:将预训练权重导入ECAResnet50加一个多层感知器构成的对抗网络,通过ECAResnet50网络提取的图像高级语义特征,计算各个高级语义特征的cosine相似度,并根据相似度的大小将相似度较大的语义特征对应的图像划分为邻居;
步骤5:将最近邻居作为先验进行聚类学习。通过学习一个聚类的函数Φη,它表示目标对应的预测类别,η表示神经网络权重参数,对数据集D中的一个样本X和它的邻居集合NX一起进行伪标签分配,在C个类别上的阈值分配,样本X被分配为第c个类的概率表示为通过目标函数Λ学习函数Φη的权重参数。目标函数Λ如下:
其中,D为数据集,X表示样本,Φη(X)表示聚类函数,η表示神经网络权重参数,<·>为点乘,λ表示,表示所有样本所属类别概率的均值,C表示目标对应类的置信度。为了让样本Xi和它的邻居产生一致的预测结果,只有它们都预测为同一个类,且都为one-hot结果时,点乘的结果最大。为了避免将所以的样本都归为同一类,又加了一个惩罚项,使得预测结果均匀分布到所有的类。具体实施中,对于K近邻中K的取值,当K=0,仅仅用样本和它们的增强图像,K>1时考虑到了类内样本的不同,但是也引入了惩罚项,因为并不是所有的邻居都属于同一个类。
步骤6:通过self-labeling进行微调。就是选择高置信度预测(pmax≈1)的样本,需要定义一个阈值,选择置信度大于阈值的样本,这样就获得了伪标签,然后就可以计算交叉熵损失来更新参数,为了避免过拟合,这里用了强增强的样本作为输入,然后不断地添加超过阈值的样本为高置信度的样本,迭代有限次后结束,分类结果得到四个类别的伪标签。
改进的YOLOv3使用Darknet-53的骨干网络对全图像进行多尺度训练。利用骨干网引入特征金字塔思想提取多尺度特征。提取三层不同尺度的特征,用于预测盒子检测不同大小的物体。对较小尺度的特征层进行上采样,通过反褶积变换到与前一特征层相同的尺寸,然后进行拼接,得到三个不同尺度的特征层之间的信息。
如图2所示,原有的YOLOv3模型的结构具体结构包括:
1.骨干网络Darknet-53。将增强后的训练样本集中的图像输入到骨干网络Darknet-53的YOLOv3网络模型中进行多尺度变换,通过骨干网Darknet-53提取多个尺度的特征;
2.特征增强及预测网络。需要Darknet-53输出的三个特征层作为输入,三个特征层位于主干部分darknet53的不同位置,分别位于中层,中下层,底层,三个特征层的大小分别为(52,52,256),(26,26,512)和(13,13,1024),然后对三个特征层进行5次卷积处理;理完后的特征层一部分用于输出该特征层对应的预测结果,一部分用于进行反卷积UmSampling2d后与上一层进行结合,之后每层特征进行一次3×3和1×1的卷积处理;对于一张图片,若初始将其划分为K×K的网格,K表示截取的图片网格数量,越大表示截取的网格尺度越小;需要三个尺度,预测C个类别,那么最终每个规模得到的张量为K×K×[3×(4+1+C)]。其中4和1分别表示预测位置的x,y坐标偏移量x_offset和y_offset,预测目标的宽h和高w,以及目标的预测类别。网络使用多个独立的逻辑回归分类器进行分类,每个分类器对于目标边框中出现的物体只判断其是否属于当前标签,即简单的二分类,这样便实现了多标签分类。
3.解码部分。通过将每个网格点加上它对应的x_offset和y_offset,加完后的结果就是预测框的中心,然后再利用先验框和h、w结合,计算出预测框的长和宽,然后通过位置信息与实际标注信息计算定位损失,通过预测类别信息和实际标注类别信息计算分类损失,过程如下:
1.判断真实框在图片中的位置,判断其属于哪一个网格点去检测;
2.判断真实框和哪个先验框重合程度最高;
3.计算该网格点应该有怎么样的预测结果才能获得真实框;
4.对所有真实框进行如上处理;
5.获得网络应该有的预测结果,将其与实际的预测结果对比。
根据分类损失和定位损失,当损失收敛到一定程度,不在下降或者达到一定的迭代次数时,完成模型的训练。
名为Darknet-53的骨干网络对全图像进行多尺度训练。Darknet-53包括五个大的残差卷积块,分别包含1、2、8、8和4个小的残差卷积单元,Darknet-53使用了残差网络Residual,darknet53中的残差卷积就是进行一次3X3、步长为2的卷积,然后保存该卷积layer,再进行一次1X1的卷积和一次3X3的卷积,并把这个结果加上layer作为最后的结果,残差网络的特点是容易优化,并且能够通过增加相当的深度来提高准确率,其内部的残差单元使用了跳跃连接,缓解了在深度神经网络中增加深度带来的梯度消失问题。darknet53的每一个卷积部分使用了特有的DarknetConv2D结构,每一次卷积的时候进行l2正则化,完成卷积后进行BatchNormalization标准化与LeakyReLU。普通的ReLU是将所有的负值都设为零,Leaky ReLU则是给所有负值赋予一个非零斜率。以数学的方式可以表示为:
其中,xi表示输入的标准化数据,ai是一个自定义的尺度放缩值,它可以将数据调整到接近0的值,yi表示激活函数的输出。
利用骨干网Darknet-53引入特征金字塔思想提取多尺度特征;提取三层不同尺度的特征,用于预测盒子和检测不同大小的物体;对较小尺度的特征层进行上采样,通过反卷积变换到与前一特征层相同的尺寸,然后进行拼接。这样,三个不同尺度的特征层之间的信息就可以被获取。
如图3所示,ECANet模块在不降维的情况下利用全局平均池化聚集卷积特性后,首先自适应确定卷积核的大小K,然后进行一维卷积,再通过sigmoid函数学习通道注意力。由于采用可视化通道特征不能高效的捕获所有通道之间的依赖关系,因此,ECANet只考虑当前信道与其k邻域通道之间的信息交换。每个通道的参数为C,则参数为k×C。各通道的参数的公式为:
该策略可以通过一维卷积的方式简洁快速地实现,其核大小为k,处理的公式为:
ω=σ(C1Dk(y)),
其中,C1D表示一维卷积,ECANet模块通过调用上述公式,使最终的参数数为k。
通过在Darknet-53残差卷积块的末尾添加轻量级注意力机制ECANet模块,得到提取两个尺度细粒度特征的ECA_Darknet-53主干。因为只需要在实际应用场景中检测大、中尺度的人脸,所以提取ECA_Darknet-53最后两个尺度的特征,然后进行进一步的特征处理。同时更大的物体可以被进一步处理以完成新的任务。
如图3所示,ECANet注意力机制模块中,一个大小为W×H×C的卷积块,给定使用全局平均池(GAP)的聚合特性,通过执行大小为k的快速一维卷积来生成通道权值,其中k通过通道维C的函数自适应地确定。
如图4所示,本发明提出的口罩佩戴视频检测任务的主要网络结构包括:数据增强网络,对训练数据集进行数据增强;ECADarknet-53通过在Darknet-53每个小的残差卷积单元的末尾加入一个ECANet注意力机制模块得到,引入ECANet注意力机制模块的骨干网ECADarknet-53可以更好的提升网络提取对任务更相关的特征的能力;删减后续原有YOLOv3网络中提取的三个不同尺度特征层中最小尺度的特征层,使得网络更加专注大中尺度的目标,同时网络更加精简,提升检测速度,如图4的two layers of features部分所示,在通过主干网后三个大的残差快提取特征后,只输出最后的两个大的残差快提取的特征,两个特征层的shape分别为(26,26,512)、(13,13,1024),它们包含了更大尺度的特征;最后一个特征层进行5次卷积处理后一部分用于输出该特征层对应的预测结果,一部分用于进行反卷积UmSampling2d后与上一个特征层进行结合,进行5次卷积处理输出对应的预测结果。
在一个实施例中,通过改进的SCAN图像无监督自分类方法对不规范佩戴口罩数据集标签重新标注,进行子类划分,得到最终的训练和测试数据集,基于改进的YOLOv3的口罩佩戴情况视频检测方法,通过YOLOv3自带的数据增强方法,结合迁移学习进行训练,得到最终的检测模型。硬件设备采用海康威视人体测温双光筒机(DS-2TD2637B-10)作为图像采集设备,布控采用三脚架配合简易安装;结合一台显卡为GeForce GTX 1060Ti的台式电脑。如图5所示,视频采集配有一个硬盘录像机,并通过一个交换机连接,实现数据传输。
模型的损失函数的表达式为:
其中,λcoord和λnoobj为对应项的权重,S2表示网格个数,B表示每个网格产生候选框的个数,表示第i个网格的第j个深度学习目标检查算法的锚框anchor box是否负责预测这个对象,表示第i个网格的第j个anchor box不负责预测这个对象,xi表示第i个网格实际中心点的横坐标,yi表示第i个网格实际中心点的纵坐标,第i个网格的第j个anchor box预测并解码后的中心点横坐标,表示第i个网格的第j个anchor box预测并解码后的中心点纵坐标,ω、h分别表示目标的宽和高,分别表示解码后目标的宽和高,C表示目标预测框内含有目标物体的置信度,表示解码后目标预测框内含有目标物体的置信度,classes表示数据集的所有类别,P表示目标属于类别c的概率,表示解释目标属于类别c的概率。
最终的实施例可以实现快速准确的大、中尺度人脸口罩佩戴情况识别和体温监控。
以上所举实施例,对本发明的目的、技术方案和优点进行了进一步的详细说明,所应理解的是,以上所举实施例仅为本发明的优选实施方式而已,并不用以限制本发明,凡在本发明的精神和原则之内对本发明所作的任何修改、等同替换、改进等,均应包含在本发明的保护范围之内。
Claims (8)
1.一种基于Slim-YOLOv3的口罩佩戴情况检测方法,其特征在于,包括:实时获取人脸视频数据,对人脸视频数据进行预处理;将预处理后的人脸图像输入到训练好的改进Slim-YOLOv3模型中,判断用户是否正确佩戴口罩;所述改进的Slim-YOLOv3模型包括骨干网络ECADarknet-53、特征增强及预测网络以及解码网络;
训练改进的Slim-YOLOv3模型的过程包括:
S1:获取原始数据集,对原始数据集进行预处理,得到训练样本集和测试样本集;
S2:对训练样本集和测试样本集中的数据进行分类和重新标注处理;
S3:将分类后的训练样本集输入到骨干网络Darknet-53中进行多尺度变换,提取多个尺度特征;
S4:将多个尺度特征输入到特征增强及预测网络中,得到分类预测结果;
S5:将分类预测结果输入到解码网络中进行解码;
S51:将每个网格点加上对应的x_offset和y_offset,得到预测框的中心;其中x_offset和y_offset分别表示网格左上角坐标(x,y)和实际预测的点x和y方向上的偏移量;
S52:将先验框和h、w结合,并计算出预测框的长和宽;其中h,w分别表示预测框的尺度缩放值;
S53:通过位置信息与实际标注信息计算定位损失,通过预测类别信息和实际标注类别信息计算分类损失;
S54:判断真实框在图片中的位置,判断其属于哪一个网格点去检测;
S55:计算真实框与先验框重合程度,并选取重合度最高的先验框进行验证;
S56:获得网络应该有的预测结果,将其与实际的标注结果对比;
S6:根据解码结果计算模型的损失函数;
其中,λcoord和λnoobj为对应项的权重,S2表示网格个数,B表示每个网格产生候选框的个数,表示第i个网格的第j个深度学习目标检查算法的锚框anchor box是否负责预测这个对象,表示第i个网格的第j个anchor box不负责预测这个对象,xi表示第i个网格实际中心点的横坐标,yi表示第i个网格实际中心点的纵坐标,第i个网格的第j个anchorbox预测并解码后的中心点横坐标,表示第i个网格的第j个anchor box预测并解码后的中心点纵坐标,ω、h分别表示目标的宽和高,分别表示解码后目标的宽和高,C表示目标预测框内含有目标物体的置信度,表示解码后目标预测框内含有目标物体的置信度,classes表示数据集的所有类别,P表示目标属于类别c的概率,表示解释目标属于类别c的概率;
S7:将测试集中的数据输入到模型进行预测,根据预测结果对模型的损失函数进行优化,当损失函数变化较小或达到迭代次数时,完成模型的训练。
2.根据权利要求1所述的一种基于Slim-YOLOv3的口罩佩戴情况检测方法,其特征在于,对原始数据集进行预处理包括:将原始数据集中的数据进行压缩、翻转以及改变图像的明暗程度,得到增强的图像数据;对增强后的图像数据进行分割,得到训练样本集和测试样本集。
3.根据权利要求1所述的一种基于Slim-YOLOv3的口罩佩戴情况检测方法,其特征在于,对训练样本集和测试样本集中的数据进行分类的过程包括:根据原始数据集的图像人脸佩戴口罩情况划分为三类,包括规范佩戴口罩图、不规范佩戴口罩图以及不佩戴口罩图;采用改进的图像无监督自分类方法SCAN对不规范佩戴口罩图进行再次分类,得到了多个子类。
4.根据权利要求3所述的一种基于Slim-YOLOv3的口罩佩戴情况检测方法,其特征在于,采用改进的图像无监督自分类方法SCAN对不规范佩戴口罩图进行分类的过程包括:
步骤1:提取口罩数据集中不规范佩戴口罩的人脸区域作为训练集;
步骤2:采用ECAResnet50网络对口罩佩戴情况数据集人脸区域数据进行分类训练,得到预训练权重;
步骤3:将预训练权重导入ECAResnet50的对抗网络中,提取图像的高级语义特征;
步骤4:计算各个高级语义特征的cosine相似度,并将相似度较大的语义特征对应的图像划分为邻居;
步骤5:将最近邻居作为先验进行聚类学习;
步骤6:采用self-labeling标签对聚类学习的图像进行微调标记处理,得到四个类别的伪标签。
6.根据权利要求5所述的一种基于Slim-YOLOv3的口罩佩戴情况检测方法,其特征在于,采用骨干网络ECADarknet-53提取分类后的训练样本集中图像的多尺度特征的过程包括:将图像输入到数据增强模块中,将图像调整为416*416*3大小;将调整后的图像输入到ECADarknet53网络中,采用一个卷积快对图像进行一次卷积升维,得到一个batch_size*416*416*32大小的图像;采用引入注意力机制ECANet模块的五个残差卷积块对卷积升维后的图形进行特征提取,每经过一个残差卷积块,所提取的特征尺度变大,最终输出第四残差卷积块和第五残差卷积块得到的两个特征层;其中,batch_size表示每次输入网络的图像数量。
7.根据权利要求6所述的一种基于Slim-YOLOv3的口罩佩戴情况检测方法,其特征在于,注意力机制ECANet模块对特征进行处理的过程包括:采用不降低维数的通道级对特征层进行全局平均池化操作;每个通道选取k个近邻通道的数据对全局平均池化操作后的数据进行1×1的卷积,并通过一个sigmod激活函数;将激活后的数据扩展到输入特征大小并与输入特征相乘,得到含有多个通道信息的增强特征。
8.根据权利要求1所述的一种基于Slim-YOLOv3的口罩佩戴情况检测方法,其特征在于,采用特征增强及预测网络对多个尺度特征进行处理的过程包括:
步骤1:将ECADarknet53网络中的第五残差卷积块得到的特征进行五次卷积处理;
步骤2:将进行卷积处理后的特征再次进行一次3×3卷积和一次1×1的卷积处理,将该处理的结果作为第五残差卷积块对应尺度特征层的预测结果;
步骤3:对五次卷积处理后的特征进行反卷积UmSampling2d操作后与第四残差卷积块得到的特征层进行堆叠处理,融合增强两个尺度特征的信息;
步骤4:对融合特征图进行五次卷积处理,将进行五次卷积的特征图进行一次3×3卷积和一次1×1的卷积,得到第四残差卷积后块对应尺度特征层的预测结果;
步骤5:输出两个尺度的特征层的预测结果,每个尺度的预测结果包括两个先验框每个网格点对应的预测框及其种类,即两个特征层分别对应着图片被分为不同大小的网格后,每个网格点上三个先验框对应的位置、置信度及其种类。
Priority Applications (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202110330611.2A CN112949572B (zh) | 2021-03-26 | 2021-03-26 | 基于Slim-YOLOv3的口罩佩戴情况检测方法 |
Applications Claiming Priority (1)
Application Number | Priority Date | Filing Date | Title |
---|---|---|---|
CN202110330611.2A CN112949572B (zh) | 2021-03-26 | 2021-03-26 | 基于Slim-YOLOv3的口罩佩戴情况检测方法 |
Publications (2)
Publication Number | Publication Date |
---|---|
CN112949572A CN112949572A (zh) | 2021-06-11 |
CN112949572B true CN112949572B (zh) | 2022-11-25 |
Family
ID=76227145
Family Applications (1)
Application Number | Title | Priority Date | Filing Date |
---|---|---|---|
CN202110330611.2A Active CN112949572B (zh) | 2021-03-26 | 2021-03-26 | 基于Slim-YOLOv3的口罩佩戴情况检测方法 |
Country Status (1)
Country | Link |
---|---|
CN (1) | CN112949572B (zh) |
Families Citing this family (12)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN113553936A (zh) * | 2021-07-19 | 2021-10-26 | 河北工程大学 | 一种基于改进型YOLOv3的口罩佩戴检测方法 |
CN113516194B (zh) * | 2021-07-20 | 2023-08-08 | 海南长光卫星信息技术有限公司 | 高光谱遥感影像半监督分类方法、装置、设备及存储介质 |
CN113553984B (zh) * | 2021-08-02 | 2023-10-13 | 中再云图技术有限公司 | 一种基于上下文辅助的视频口罩检测方法 |
CN113762201B (zh) * | 2021-09-16 | 2023-05-09 | 深圳大学 | 基于yolov4的口罩检测方法 |
CN113989708B (zh) * | 2021-10-27 | 2024-06-04 | 福州大学 | 一种基于YOLO v4的校园图书馆疫情防控方法 |
CN114283462B (zh) * | 2021-11-08 | 2024-04-09 | 上海应用技术大学 | 口罩佩戴检测方法及*** |
CN114092998A (zh) * | 2021-11-09 | 2022-02-25 | 杭州电子科技大学信息工程学院 | 一种基于卷积神经网络的佩戴口罩人脸识别检测方法 |
CN114118061A (zh) * | 2021-11-30 | 2022-03-01 | 深圳市北科瑞声科技股份有限公司 | 轻量级意图识别模型训练方法、装置、设备及存储介质 |
CN114155453B (zh) * | 2022-02-10 | 2022-05-10 | 深圳爱莫科技有限公司 | 冰柜商品图像识别训练方法、识别方法及占有率计算方法 |
CN114821702A (zh) * | 2022-03-15 | 2022-07-29 | 电子科技大学 | 一种基于遮挡人脸的热红外人脸识别方法 |
CN116311104B (zh) * | 2023-05-15 | 2023-08-22 | 合肥市正茂科技有限公司 | 一种车辆改装识别模型的训练方法、装置、设备及介质 |
CN117975376B (zh) * | 2024-04-02 | 2024-06-07 | 湖南大学 | 基于深度分级融合残差网络的矿山作业安全检测方法 |
Citations (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN111626330A (zh) * | 2020-04-23 | 2020-09-04 | 南京邮电大学 | 基于多尺度特征图重构和知识蒸馏的目标检测方法与*** |
CN111881775A (zh) * | 2020-07-07 | 2020-11-03 | 烽火通信科技股份有限公司 | 一种人脸实时识别方法和装置 |
CN112183471A (zh) * | 2020-10-28 | 2021-01-05 | 西安交通大学 | 一种现场人员防疫口罩规范佩戴的自动检测方法及*** |
Family Cites Families (2)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN111862408B (zh) * | 2020-06-16 | 2022-07-05 | 国能信控互联技术有限公司 | 一种智能门禁控制方法 |
AU2020101210A4 (en) * | 2020-06-30 | 2020-08-06 | Anguraj, Dinesh Kumar Dr | Automated screening system of covid-19 infected persons by measurement of respiratory data through deep facial recognition |
-
2021
- 2021-03-26 CN CN202110330611.2A patent/CN112949572B/zh active Active
Patent Citations (3)
Publication number | Priority date | Publication date | Assignee | Title |
---|---|---|---|---|
CN111626330A (zh) * | 2020-04-23 | 2020-09-04 | 南京邮电大学 | 基于多尺度特征图重构和知识蒸馏的目标检测方法与*** |
CN111881775A (zh) * | 2020-07-07 | 2020-11-03 | 烽火通信科技股份有限公司 | 一种人脸实时识别方法和装置 |
CN112183471A (zh) * | 2020-10-28 | 2021-01-05 | 西安交通大学 | 一种现场人员防疫口罩规范佩戴的自动检测方法及*** |
Non-Patent Citations (1)
Title |
---|
YOLOv3_slim for face mask recognition;Jiang Xiaoming et al;《Journal of Physics: Conference Series》;20210201;全文 * |
Also Published As
Publication number | Publication date |
---|---|
CN112949572A (zh) | 2021-06-11 |
Similar Documents
Publication | Publication Date | Title |
---|---|---|
CN112949572B (zh) | 基于Slim-YOLOv3的口罩佩戴情况检测方法 | |
CN113361495B (zh) | 人脸图像相似度的计算方法、装置、设备及存储介质 | |
Yaakob et al. | An insect classification analysis based on shape features using quality threshold ARTMAP and moment invariant | |
CN113159120A (zh) | 一种基于多尺度跨图像弱监督学习的违禁物检测方法 | |
Mai et al. | Comparing salient object detection results without ground truth | |
CN110929593A (zh) | 一种基于细节辨别区别的实时显著性行人检测方法 | |
CN106971158B (zh) | 一种基于CoLBP共生特征与GSS特征的行人检测方法 | |
CN110222636B (zh) | 基于背景抑制的行人属性识别方法 | |
Lingappa et al. | Medical image segmentation based on extreme learning machine algorithm in kernel fuzzy c-means using artificial bee colony method | |
Nercessian et al. | Automatic detection of potential threat objects in X-ray luggage scan images | |
CN106599864A (zh) | 一种基于极值理论的深度人脸识别方法 | |
Teimouri et al. | A real-time ball detection approach using convolutional neural networks | |
CN116311412A (zh) | 一种融合3d注意力机制和空洞卷积的口罩佩戴检测方法 | |
CN111274964A (zh) | 一种基于无人机视觉显著性分析水面污染物的检测方法 | |
Deeksha et al. | Classification of Brain Tumor and its types using Convolutional Neural Network | |
CN109902692A (zh) | 一种基于局部区域深度特征编码的图像分类方法 | |
CN105844299B (zh) | 一种基于词袋模型的图像分类方法 | |
CN111738194A (zh) | 一种用于人脸图像相似性的评价方法和装置 | |
Nanthini et al. | A novel Deep CNN based LDnet model with the combination of 2D and 3D CNN for Face Liveness Detection | |
CN116434010A (zh) | 一种多视图的行人属性识别方法 | |
Lulio et al. | Jseg algorithm and statistical ann image segmentation techniques for natural scenes | |
Mahmood | Defocus Blur Segmentation Using Genetic Programming and Adaptive Threshold. | |
CN112287929A (zh) | 基于特征集成深度学习网络的遥感图像显著性分析方法 | |
Jensch et al. | A comparative evaluation of three skin color detection approaches | |
Mondal et al. | Anomaly detection using context dependent optical flow |
Legal Events
Date | Code | Title | Description |
---|---|---|---|
PB01 | Publication | ||
PB01 | Publication | ||
SE01 | Entry into force of request for substantive examination | ||
SE01 | Entry into force of request for substantive examination | ||
GR01 | Patent grant | ||
GR01 | Patent grant |