一种人脸检测模型训练方法、人脸检测方法及其相关装置与流程

文档序号:25886313发布日期:2021-07-16 19:17阅读:89来源:国知局
一种人脸检测模型训练方法、人脸检测方法及其相关装置与流程

1.本申请涉及人脸检测技术领域,尤其涉及一种人脸检测模型训练方法、人脸检测方法及其相关装置。


背景技术:

2.人脸检测是自动识别系统中的一个重要环节,其包括人脸框检测、人脸关键点检测,现有技术需要采用两个网络模型进行人脸框检测和人脸关键点检测,使得模型使用效率较低;且在一个设备中嵌入多个网络模型进行人脸检测,会影响设备的检测速度。若简单的将人脸框检测、人脸关键点检测集成在一个网络模型中,检测精度难以保证。
3.因此,提供一种高精度的人脸检测方法是目前亟需解决的技术问题。


技术实现要素:

4.本申请提供了一种人脸检测模型训练方法、人脸检测方法及其相关装置,用于解决现有技术在进行人脸框检测和人脸关键点检测时,需采用两个网络模型,存在的模型使用效率低和检测速度慢的技术问题。
5.有鉴于此,本申请第一方面提供了一种人脸检测模型训练方法,包括:
6.构建人脸检测网络,所述人脸检测网络包括依次连接的主干网络、转置卷积模块和预测模块,所述主干网络中的若干个卷积模块与所述转置卷积模块跳跃连接;
7.获取人脸训练样本,所述人脸训练样本的标签信息包括真实人脸关键点坐标和真实人脸检测框;
8.将所述人脸训练样本输入到所述人脸检测网络,使得所述主干网络对所述人脸训练样本进行卷积处理,所述转置卷积模块对所述主干网络提取的目标卷积特征图进行上采样处理,得到采样特征图,所述预测模块对所述采样特征图分别进行人脸关键点预测、人脸关键点偏置预测和人脸检测框预测,得到所述人脸训练样本的人脸预测结果,其中,所述目标卷积特征图包括所述主干网络最后一个卷积模块输出的卷积特征图,以及与所述转置卷积模块跳跃连接的卷积模块输出的卷积特征图;
9.基于所述人脸训练样本的人脸预测结果和所述标签信息计算网络损失值,并通过所述网络损失值更新所述人脸检测网络的参数,直至所述人脸检测网络收敛,得到人脸检测模型。
10.可选的,所述主干网络为采用空洞卷积层代替mobilenetv3网络浅层中的深度可分离卷积层得到的改进后的mobilenetv3网络,所述主干网络的第五个卷积模块的输出端与所述转置卷积模块的第一个转置卷积层的输出端跳跃连接,所述主干网络的第四个卷积模块的输出端与所述转置卷积模块的第二个转置卷积层的输出端跳跃连接。
11.可选的,所述人脸训练样本的人脸预测结果包括人脸关键点预测结果、人脸关键点偏置预测结果和人脸检测框预测结果,所述基于所述人脸训练样本的人脸预测结果和所述标签信息计算网络损失值,包括:
12.根据所述人脸训练样本的所述真实人脸关键点坐标和所述人脸关键点预测结果,通过人脸关键点损失函数计算人脸关键点损失值;
13.根据所述人脸训练样本的所述真实人脸检测框和所述人脸检测框预测结果,通过人脸尺寸损失函数计算人脸尺寸损失值;
14.根据所述人脸训练样本的所述人脸关键点偏置预测结果,以及所述真实人脸检测框的中心点和所述人脸检测框预测结果的中心点的差值,通过人脸关键点偏置损失函数计算人脸关键点偏置损失值;
15.根据预置权重对所述人脸关键点损失值、所述人脸关键点偏置损失值和所述人脸尺寸损失值进行加权求和,得到最终的网络损失值。
16.本申请第二方面提供了一种人脸检测方法,包括:
17.获取待检测人脸图像;
18.将所述待检测人脸图像输入到人脸检测模型进行人脸关键点预测和人脸检测框预测,得到所述待检测人脸图像的人脸预测结果,其中,所述人脸检测模型通过权利要求第一方面任一种所述的人脸检测模型训练方法得到。
19.本申请第三方面提供了一种人脸检测模型训练装置,包括:
20.构建单元,用于构建人脸检测网络,所述人脸检测网络包括依次连接的主干网络、转置卷积模块和预测模块,所述主干网络中的若干个卷积模块与所述转置卷积模块跳跃连接;
21.获取单元,用于获取人脸训练样本,所述人脸训练样本的标签信息包括真实人脸关键点坐标和真实人脸检测框;
22.输入单元,用于将所述人脸训练样本输入到所述人脸检测网络,使得所述主干网络对所述人脸训练样本进行卷积处理,所述转置卷积模块对所述主干网络提取的目标卷积特征图进行上采样处理,得到采样特征图,所述预测模块对所述采样特征图分别进行人脸关键点预测、人脸关键点偏置预测和人脸检测框预测,得到所述人脸训练样本的人脸预测结果,其中,所述目标卷积特征图包括所述主干网络最后一个卷积模块输出的卷积特征图,以及与所述转置卷积模块跳跃连接的卷积模块输出的卷积特征图;
23.计算单元,用于基于所述人脸训练样本的人脸预测结果和所述标签信息计算网络损失值,并通过所述网络损失值更新所述人脸检测网络的参数,直至所述人脸检测网络收敛,得到人脸检测模型。
24.可选的,所述主干网络为采用空洞卷积层代替mobilenetv3网络浅层中的深度可分离卷积层得到的改进后的mobilenetv3网络,所述主干网络的第五个卷积模块的输出端与所述转置卷积模块的第一个转置卷积层的输出端跳跃连接,所述主干网络的第四个卷积模块的输出端与所述转置卷积模块的第二个转置卷积层的输出端跳跃连接。
25.可选的,所述人脸训练样本的人脸预测结果包括人脸关键点预测结果、人脸关键点偏置预测结果和人脸检测框预测结果,所述计算单元具体用于:
26.根据所述人脸训练样本的所述真实人脸关键点坐标和所述人脸关键点预测结果,通过人脸关键点损失函数计算人脸关键点损失值;
27.根据所述人脸训练样本的所述真实人脸检测框和所述人脸检测框预测结果,通过人脸尺寸损失函数计算人脸尺寸损失值;
28.根据所述人脸训练样本的所述人脸关键点偏置预测结果,以及所述真实人脸检测框的中心点和所述人脸检测框预测结果的中心点的差值,通过人脸关键点偏置损失函数计算人脸关键点偏置损失值;
29.根据预置权重对所述人脸关键点损失值、所述人脸关键点偏置损失值和所述人脸尺寸损失值进行加权求和,得到最终的网络损失值;
30.通过所述网络损失值更新所述人脸检测网络的参数,直至所述人脸检测网络收敛,得到人脸检测模型。
31.本申请第三方面提供了一种人脸检测装置,包括:
32.获取单元,用于获取待检测人脸图像;
33.预测单元,用于将所述待检测人脸图像输入到人脸检测模型进行人脸关键点预测和人脸检测框预测,得到所述待检测人脸图像的人脸预测结果,其中,所述人脸检测模型通过第一方面任一种所述的人脸检测模型训练方法得到。
34.本申请第四方面提供了一种设备,包括处理器以及存储器;
35.所述存储器用于存储程序代码,并将所述程序代码传输给所述处理器;
36.所述处理器用于根据所述程序代码中的指令执行第一方面任一种所述的人脸检测模型训练方法,或执行第二方面所述的人脸检测方法。
37.本申请第五方面提供了一种计算机可读存储介质,所述计算机可读存储介质用于存储程序代码,所述程序代码用于执行第一方面任一种所述的人脸检测模型训练方法,或执行第二方面所述的人脸检测方法。
38.从以上技术方案可以看出,本申请具有以下优点:
39.本申请提供了一种人脸检测模型训练方法,包括:构建人脸检测网络,人脸检测网络包括依次连接的主干网络、转置卷积模块和预测模块,主干网络中的若干个卷积模块与转置卷积模块跳跃连接;获取人脸训练样本,人脸训练样本的标签信息包括真实人脸关键点坐标和真实人脸检测框;将人脸训练样本输入到人脸检测网络,使得主干网络对人脸训练样本进行卷积处理,转置卷积模块对主干网络提取的目标卷积特征图进行上采样处理,得到采样特征图,预测模块对采样特征图分别进行人脸关键点预测、人脸关键点偏置预测和人脸检测框预测,得到人脸训练样本的人脸预测结果,其中,目标卷积特征图包括主干网络最后一个卷积模块输出的卷积特征图,以及与转置卷积模块跳跃连接的卷积模块输出的卷积特征图;基于人脸训练样本的人脸预测结果和标签信息计算网络损失值,并通过网络损失值更新人脸检测网络的参数,直至人脸检测网络收敛,得到人脸检测模型。
40.本申请中,构建人脸检测网络时,在主干网络后增加一个转置卷积模块,通过转置卷积模块还原待检测人脸图像中的人脸的位置和语义信息,通过将主干网络中的卷积模块与转置卷积模块进行跳跃连接,通过跳跃连接来弥补上采样过程中丢失的细节数据,提高人脸检测准确率;通过预测模块进行人脸关键点预测、人脸关键点偏置预测和人脸检测框预测,通过人脸关键点偏置预测结果修正人脸关键点预测结果,进一步提高人脸检测结果,本申请通过一个模型实现了人脸检测和人脸关键点检测,提高了模型使用效率,解决了现有技术在进行人脸框检测和人脸关键点检测时,需采用两个网络模型,存在的模型使用效率低和检测速度慢的技术问题。
附图说明
41.为了更清楚地说明本申请实施例或现有技术中的技术方案,下面将对实施例或现有技术描述中所需要使用的附图作简单地介绍,显而易见地,下面描述中的附图仅仅是本申请的一些实施例,对于本领域普通技术人员来讲,在不付出创造性劳动性的前提下,还可以根据这些附图获得其它的附图。
42.图1为本申请实施例提供的一种人脸检测模型训练方法的一个流程示意图;
43.图2本申请实施例提供的一种人脸检测网络的一个结构示意图;
44.图3本申请实施例提供的三种空洞卷积核的示意图;
45.图4为本申请实施例提供的一种人脸检测方法的一个流程示意图;
46.图5为本申请实施例提供的一种人脸检测模型训练装置的一个结构示意图;
47.图6为本申请实施例提供的一种人脸检测装置的一个结构示意图。
具体实施方式
48.本申请提供了一种人脸检测模型训练方法、人脸检测方法及其相关装置,用于解决现有技术在进行人脸框检测和人脸关键点检测时,需采用两个网络模型,存在的模型使用效率低和检测速度慢的技术问题。
49.为了使本技术领域的人员更好地理解本申请方案,下面将结合本申请实施例中的附图,对本申请实施例中的技术方案进行清楚、完整地描述,显然,所描述的实施例仅是本申请一部分实施例,而不是全部的实施例。基于本申请中的实施例,本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例,都属于本申请保护的范围。
50.为了便于理解,请参阅图1,本申请提供的一种人脸检测模型训练方法的一个实施例,包括:
51.步骤101、构建人脸检测网络,人脸检测网络包括依次连接的主干网络、转置卷积模块和预测模块,主干网络中的若干个卷积模块与转置卷积模块跳跃连接。
52.本申请实施例中的人脸检测网络包括依次连接的主干网络、转置卷积模块和预测模块,主干网络中的若干个卷积模块与转置卷积模块跳跃连接,人脸检测网络的具体结构可以参考图2。为了保证人脸检测的实时性,本申请实施例中的主干网络优选采用轻量级网络。
53.步骤102、获取人脸训练样本,人脸训练样本的标签信息包括真实人脸关键点坐标和真实人脸检测框。
54.可以从现有的公开的人脸数据集中获取人脸训练样本,该人脸训练样本均标注有对应的人脸框位置和人脸关键点位置。
55.步骤103、将人脸训练样本输入到人脸检测网络,使得主干网络对人脸训练样本进行卷积处理,转置卷积模块对主干网络提取的目标卷积特征图进行上采样处理,得到采样特征图,预测模块对采样特征图分别进行人脸关键点预测、人脸关键点偏置预测和人脸检测框预测,得到人脸训练样本的人脸预测结果,其中,目标卷积特征图包括主干网络最后一个卷积模块输出的卷积特征图,以及与转置卷积模块跳跃连接的卷积模块输出的卷积特征图。
56.将人脸训练样本输入到人脸检测网络之前,可以对人脸训练样本进行预处理,例
如,裁剪、归一化、去噪等。
57.在实际应用场景中,当人脸靠近设备时,检测的人脸尺寸较大,对人脸尺寸检测和关键点检测需要较大的感受野,感受野可以看作经过卷积后每一层输出的卷积特征图上的像素点在原图像上映射的区域大小。
58.本申请实施例考虑在标准的卷积中增加空洞,用来增加卷积核的感受野,这样的卷积核又称为空洞卷积核,空洞卷积核和标准卷积核的主要不同在于卷积核之间增加了零值,如图3所示。图3的左部分图为空洞率为1的标准3
×
3卷积核的感受野,图3的中间图为空洞率为2的标准3
×
3卷积核的感受野,感受野大小为5
×
5,图3的右部分图为空洞率为3的标准3
×
3卷积核的感受野,感受野大小为7
×
7。通过图3可以发现,对于标准3
×
3卷积核添加不同数目的空洞,能够扩大感受野且不需要引入额外的参数数量。
59.卷积神经网络浅层提取的是一些边缘、条纹等信息,层数越深,提取的特征越抽象,为了避免使用空洞卷积时,产生的稀疏化的输入对卷积结果的影响,本申请实施例只考虑在主干网络的浅层使用空洞率为2的空洞卷积层代替深度可分离卷积层,这样在相同参数量和计算量下,拥有更大的感受野,提高对较大人脸图像的人脸尺寸检测和关键点检测的准确率。
60.本申请实施例中的主干网络优选采用改进后的mobilenetv3网络,包括6个卷积模块,采用空洞卷积层代替mobilenetv3网络浅层中的深度可分离卷积层,得到改进后的mobilenetv3网络。mobilenetv3网络的第一个卷积模块中的第一个深度可分离卷积层的步长为2,以避免距离过远的卷积造成局部信息的丢失,本申请实施例对其不做改进;mobilenetv3网络的第一个卷积模块中的第二个深度可分离卷积层的步长为1,本申请实施例采用空洞卷积层替换mobilenetv3网络中的第一个卷积模块(即图2中实线框内的卷积模块)中的第二个深度可分离卷积层,该空洞卷积层的空洞率为2,其他卷积模块不进行改变。
61.将人脸训练样本输入到人脸检测网络后,主干网络中的6个卷积模块依次对该待检测人脸图像进行卷积操作以提取人脸特征,得到卷积特征图。
62.本申请实施例为了更好的还原人脸图像中的人脸的位置和语义信息,在主干网络后添加转置卷积模块对主干网络提取的目标卷积特征图进行上采样处理,映射到原图大小。该转置卷积模块包括3个转置卷积层。其中,主干网络的第五个卷积模块的输出端与转置卷积模块的第一个转置卷积层的输出端跳跃连接,主干网络的第四个卷积层的输出端与转置卷积模块的第二个转置卷积层的输出端跳跃连接。由于主干网络提取的卷积特征图经过转置卷积模块的上采样处理后,得到的采样特征图是粗糙的,丢失了许多细节数据,因此,本申请实施例采用跳跃连接来弥补丢失的细节数据,改善上采样的粗糙程度,通过结合上采样和上一层卷积模块输出的卷积特征图,来修复还原的图像,有助于提高人脸检测准确率。
63.主干网络的第六个卷积模块的输出为转置卷积模块的第一个转置卷积层的输入,第一个转置卷积层对输入的卷积特征图进行上采样处理,得到第一特征图;第二个转置卷积层对该第一特征图和第五个卷积模块提取的卷积特征图进行上采样处理,得到第二特征图;第三个转置卷积层对第二特征图和第四个卷积模块输出的卷积特征图进行上采样处理,得到采样特征图。
64.转置卷积模块将提取的采样特征图送入到预测模块中进行人脸关键点预测、人脸
关键点偏置预测和人脸检测框预测,得到待检测人脸图像的人脸预测结果。
65.步骤104、基于人脸训练样本的人脸预测结果和标签信息计算网络损失值,并通过网络损失值更新人脸检测网络的参数,直至人脸检测网络收敛,得到人脸检测模型。
66.人脸训练样本的人脸预测结果包括人脸关键点预测结果、人脸关键点偏置预测结果和人脸检测框预测结果,基于人脸训练样本的人脸预测结果和标签信息计算网络损失值,包括:
67.根据人脸训练样本的真实人脸关键点坐标和人脸关键点预测结果,通过人脸关键点损失函数计算人脸关键点损失值;根据人脸训练样本的真实人脸检测框和人脸检测框预测结果,通过人脸尺寸损失函数计算人脸尺寸损失值;根据人脸训练样本的人脸关键点偏置预测结果,以及真实人脸检测框的中心点和人脸检测框预测结果的中心点的差值,通过人脸关键点偏置损失函数计算人脸关键点偏置损失值;根据预置权重对人脸关键点损失值、人脸关键点偏置损失值和人脸尺寸损失值进行加权求和,得到最终的网络损失值。
68.本申请实施例中的人脸关键点损失函数为:
[0069][0070]
式中,l
k
为人脸关键点损失值,n为输入的人脸训练样本i的关键点数量,i∈r
w
×
h
×3,w、h分别为训练样本的宽、高,α、β为超参数,可以实际情况设置具体的取值,人脸训练样本分为正、负训练样本,为人脸关键点预测值,c为人脸个数,y
xyc
为人脸训练样本的真实值(标签),y
xyc
=1表示当前像素点(x,y)是人脸关键点,y
xyc
=0表示当前像素点(x,y)不是人脸关键点。
[0071]
由于在主干网络对人脸训练样本进行特征提取时,进行了m倍的下采样,将卷积特征图重新映射到原始图像上时,会带来精度误差。因此,对每个关键点采用人脸关键点偏置损失进行补偿,人脸关键点偏置损失函数为:
[0072][0073]
式中,l
off
为人脸关键点偏置损失值,为关键点偏置预测结果,为预测的人脸检测框的中心点位置,为标注的真实人脸检测框的中心点位置p与预测的人脸检测框的中心点位置的差值。
[0074]
人脸尺寸损失函数为:
[0075][0076]
式中,l
size
为人脸尺寸损失值,假设某目标k(标注的人脸检测框)为分别为人脸检测框的左上角坐标和右下角坐标,该人脸检测框的中心点
为采用关键点中的中心点对每个目标k的尺寸进行回归,得到人脸检测框预测结果,根据人脸检测框预测结果计算得到预测的人脸尺寸人脸检测框预测结果,根据人脸检测框预测结果计算得到预测的人脸尺寸为真实的人脸尺寸。
[0077]
根据预置权重对人脸关键点损失值、人脸关键点偏置损失值和人脸尺寸损失值进行加权求和,得到最终的网络损失值l
det
,即:
[0078]
l
det
=λ
k
l
k

off
l
off

size
l
size

[0079]
式中,λ
k
、λ
off
、λ
size
分别为人脸关键点损失值、人脸关键点偏置损失值和人脸尺寸损失值的预置权重,可以根据实际情况进行设置,本申请实施例中的λ
k
优选设置为1。
[0080]
通过网络损失值更新人脸检测网络的参数,直至人脸检测网络收敛,得到人脸检测模型。本申请实施例中的人脸检测网络的损失函数中加入了人脸关键点偏置损失,对每个关键点采用人脸关键点偏置损失进行补偿,有助于提高人脸关键点检测结果。
[0081]
本申请实施例中,构建人脸检测网络时,在主干网络后增加一个转置卷积模块,通过转置卷积模块还原待检测人脸图像中的人脸的位置和语义信息,通过将主干网络中的卷积模块与转置卷积模块进行跳跃连接,通过跳跃连接来弥补上采样过程中丢失的细节数据,提高人脸检测准确率;通过预测模块进行人脸关键点预测、人脸关键点偏置预测和人脸检测框预测,通过人脸关键点偏置预测结果修正人脸关键点预测结果,进一步提高人脸检测结果,本申请实施例通过一个模型实现了人脸检测和人脸关键点检测,提高了模型使用效率,解决了现有技术在进行人脸框检测和人脸关键点检测时,需采用两个网络模型,存在的模型使用效率低和检测速度慢的技术问题。
[0082]
以上为本申请提供的一种人脸检测模型训练方法的一个实施例,以下为本申请提供的一种人脸检测方法的一个实施例。
[0083]
请参考图4,本申请实施例提供的一种人脸检测方法,包括:
[0084]
步骤201、获取待检测人脸图像。
[0085]
可以通过图像采集设备获取待检测人脸图像,将待检测人脸图像输入到人脸检测模型之前,可以对待检测人脸图像进行预处理,例如,裁剪、归一化、去噪等。
[0086]
步骤202、将待检测人脸图像输入到人脸检测模型进行人脸关键点预测和人脸检测框预测,得到待检测人脸图像的人脸预测结果。
[0087]
本申请实施例中的人脸检测模型通过前述方法实施例中的人脸检测模型训练方法得到。
[0088]
本申请实施例中,通过训练好的人脸检测模型进行人脸关键点预测和人脸检测框预测,通过一个模型实现了人脸检测和人脸关键点检测,提高了模型使用效率;通过转置卷积模块还原待检测人脸图像中的人脸的位置和语义信息,通过将主干网络中的卷积模块与转置卷积模块进行跳跃连接,提高人脸检测准确率,解决了现有技术在进行人脸框检测和人脸关键点检测时,需采用两个网络模型,存在的模型使用效率低和检测速度慢的技术问题。
[0089]
以上为本申请提供的一种人脸检测方法的一个实施例,以下为本申请提供的一种人脸检测模型训练装置的一个实施例。
[0090]
请参考图5,本申请实施例提供的一种人脸检测模型训练装置,包括:
[0091]
构建单元,用于构建人脸检测网络,人脸检测网络包括依次连接的主干网络、转置卷积模块和预测模块,主干网络中的若干个卷积模块与转置卷积模块跳跃连接;
[0092]
获取单元,用于获取人脸训练样本,人脸训练样本的标签信息包括真实人脸关键点坐标和真实人脸检测框;
[0093]
输入单元,用于将人脸训练样本输入到人脸检测网络,使得主干网络对人脸训练样本进行卷积处理,转置卷积模块对主干网络提取的目标卷积特征图进行上采样处理,得到采样特征图,预测模块对采样特征图分别进行人脸关键点预测、人脸关键点偏置预测和人脸检测框预测,得到人脸训练样本的人脸预测结果,其中,目标卷积特征图包括主干网络最后一个卷积模块输出的卷积特征图,以及与转置卷积模块跳跃连接的卷积模块输出的卷积特征图;
[0094]
计算单元,用于基于人脸训练样本的人脸预测结果和标签信息计算网络损失值,并通过网络损失值更新人脸检测网络的参数,直至人脸检测网络收敛,得到人脸检测模型。
[0095]
作为进一步地改进,主干网络为采用空洞卷积层代替mobilenetv3网络浅层中的深度可分离卷积层得到的改进后的mobilenetv3网络,主干网络的第五个卷积模块的输出端与转置卷积模块的第一个转置卷积层的输出端跳跃连接,主干网络的第四个卷积模块的输出端与转置卷积模块的第二个转置卷积层的输出端跳跃连接。
[0096]
作为进一步地改进,人脸训练样本的人脸预测结果包括人脸关键点预测结果、人脸关键点偏置预测结果和人脸检测框预测结果,计算单元具体用于:
[0097]
根据人脸训练样本的真实人脸关键点坐标和人脸关键点预测结果,通过人脸关键点损失函数计算人脸关键点损失值;
[0098]
根据人脸训练样本的真实人脸检测框和人脸检测框预测结果,通过人脸尺寸损失函数计算人脸尺寸损失值;
[0099]
根据人脸训练样本的人脸关键点偏置预测结果,以及真实人脸检测框的中心点和人脸检测框预测结果的中心点的差值,通过人脸关键点偏置损失函数计算人脸关键点偏置损失值;
[0100]
根据预置权重对人脸关键点损失值、人脸关键点偏置损失值和人脸尺寸损失值进行加权求和,得到最终的网络损失值;
[0101]
通过网络损失值更新人脸检测网络的参数,直至人脸检测网络收敛,得到人脸检测模型。
[0102]
本申请实施例中,构建人脸检测网络时,在主干网络后增加一个转置卷积模块,通过转置卷积模块还原待检测人脸图像中的人脸的位置和语义信息,通过将主干网络中的卷积模块与转置卷积模块进行跳跃连接,通过跳跃连接来弥补上采样过程中丢失的细节数据,提高人脸检测准确率;通过预测模块进行人脸关键点预测、人脸关键点偏置预测和人脸检测框预测,通过人脸关键点偏置预测结果修正人脸关键点预测结果,进一步提高人脸检测结果,本申请实施例通过一个模型实现了人脸检测和人脸关键点检测,提高了模型使用效率,解决了现有技术在进行人脸框检测和人脸关键点检测时,需采用两个网络模型,存在的模型使用效率低和检测速度慢的技术问题。
[0103]
以上为本申请提供的一种人脸检测模型训练装置的一个实施例,以下为本申请提
供的一种人脸检测装置的一个实施例。
[0104]
请参考图6,本申请实施例提供的一种人脸检测装置,包括:
[0105]
获取单元,用于获取待检测人脸图像;
[0106]
预测单元,用于将待检测人脸图像输入到人脸检测模型进行人脸关键点预测和人脸检测框预测,得到待检测人脸图像的人脸预测结果,其中,人脸检测模型通过前述方法实施例中的人脸检测模型训练方法得到。
[0107]
本申请实施例中,通过训练好的人脸检测模型进行人脸关键点预测和人脸检测框预测,通过一个模型实现了人脸检测和人脸关键点检测,提高了模型使用效率;通过转置卷积模块还原待检测人脸图像中的人脸的位置和语义信息,通过将主干网络中的卷积模块与转置卷积模块进行跳跃连接,提高人脸检测准确率,解决了现有技术在进行人脸框检测和人脸关键点检测时,需采用两个网络模型,存在的模型使用效率低和检测速度慢的技术问题。
[0108]
本申请实施例还提供一种设备,包括处理器以及存储器;
[0109]
存储器用于存储程序代码,并将程序代码传输给处理器;
[0110]
处理器用于根据程序代码中的指令执行上述人脸检测模型训练方法实施例中的人脸检测模型训练方法,或执行上述人脸检测方法实施例中的人脸检测方法。
[0111]
本申请实施例还提供了一种计算机可读存储介质,计算机可读存储介质用于存储程序代码,程序代码用于执行上述人脸检测模型训练方法实施例中的人脸检测模型训练方法,或执行上述人脸检测方法实施例中的人脸检测方法。
[0112]
所属领域的技术人员可以清楚地了解到,为描述的方便和简洁,上述描述的装置和单元的具体工作过程,可以参考前述方法实施例中的对应过程,在此不再赘述。
[0113]
在本申请所提供的几个实施例中,应该理解到,所揭露的装置和方法,可以通过其它的方式实现。例如,以上所描述的装置实施例仅仅是示意性的,例如,所述单元的划分,仅仅为一种逻辑功能划分,实际实现时可以有另外的划分方式,例如多个单元或组件可以结合或者可以集成到另一个系统,或一些特征可以忽略,或不执行。另一点,所显示或讨论的相互之间的耦合或直接耦合或通信连接可以是通过一些接口,装置或单元的间接耦合或通信连接,可以是电性,机械或其它的形式。
[0114]
所述作为分离部件说明的单元可以是或者也可以不是物理上分开的,作为单元显示的部件可以是或者也可以不是物理单元,即可以位于一个地方,或者也可以分布到多个网络单元上。可以根据实际的需要选择其中的部分或者全部单元来实现本实施例方案的目的。
[0115]
另外,在本申请各个实施例中的各功能单元可以集成在一个处理单元中,也可以是各个单元单独物理存在,也可以两个或两个以上单元集成在一个单元中。上述集成的单元既可以采用硬件的形式实现,也可以采用软件功能单元的形式实现。
[0116]
所述集成的单元如果以软件功能单元的形式实现并作为独立的产品销售或使用时,可以存储在一个计算机可读取存储介质中。基于这样的理解,本申请的技术方案本质上或者说对现有技术做出贡献的部分或者该技术方案的全部或部分可以以软件产品的形式体现出来,该计算机软件产品存储在一个存储介质中,包括若干指令用以通过一台计算机设备(可以是个人计算机,服务器,或者网络设备等)执行本申请各个实施例所述方法的全
部或部分步骤。而前述的存储介质包括:u盘、移动硬盘、只读存储器(英文全称:read

only memory,英文缩写:rom)、随机存取存储器(英文全称:random access memory,英文缩写:ram)、磁碟或者光盘等各种可以存储程序代码的介质。
[0117]
以上所述,以上实施例仅用以说明本申请的技术方案,而非对其限制;尽管参照前述实施例对本申请进行了详细的说明,本领域的普通技术人员应当理解:其依然可以对前述各实施例所记载的技术方案进行修改,或者对其中部分技术特征进行等同替换;而这些修改或者替换,并不使相应技术方案的本质脱离本申请各实施例技术方案的精神和范围。
当前第1页1 2 3 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1