[发明专利]一种人脸检测模型训练方法、人脸检测方法及其相关装置

权利要求书

1.一种人脸检测模型训练方法，其特征在于，包括：

构建人脸检测网络，所述人脸检测网络包括依次连接的主干网络、转置卷积模块和预测模块，所述主干网络中的若干个卷积模块与所述转置卷积模块跳跃连接；

获取人脸训练样本，所述人脸训练样本的标签信息包括真实人脸关键点坐标和真实人脸检测框；

将所述人脸训练样本输入到所述人脸检测网络，使得所述主干网络对所述人脸训练样本进行卷积处理，所述转置卷积模块对所述主干网络提取的目标卷积特征图进行上采样处理，得到采样特征图，所述预测模块对所述采样特征图分别进行人脸关键点预测、人脸关键点偏置预测和人脸检测框预测，得到所述人脸训练样本的人脸预测结果，其中，所述目标卷积特征图包括所述主干网络最后一个卷积模块输出的卷积特征图，以及与所述转置卷积模块跳跃连接的卷积模块输出的卷积特征图；

基于所述人脸训练样本的人脸预测结果和所述标签信息计算网络损失值，并通过所述网络损失值更新所述人脸检测网络的参数，直至所述人脸检测网络收敛，得到人脸检测模型。

2.根据权利要求1所述的人脸检测方法，其特征在于，所述主干网络为采用空洞卷积层代替MobileNetV3网络浅层中的深度可分离卷积层得到的改进后的MobileNetV3网络，所述主干网络的第五个卷积模块的输出端与所述转置卷积模块的第一个转置卷积层的输出端跳跃连接，所述主干网络的第四个卷积模块的输出端与所述转置卷积模块的第二个转置卷积层的输出端跳跃连接。

3.根据权利要求1所述的人脸检测方法，其特征在于，所述人脸训练样本的人脸预测结果包括人脸关键点预测结果、人脸关键点偏置预测结果和人脸检测框预测结果，所述基于所述人脸训练样本的人脸预测结果和所述标签信息计算网络损失值，包括：

根据所述人脸训练样本的所述真实人脸关键点坐标和所述人脸关键点预测结果，通过人脸关键点损失函数计算人脸关键点损失值；

根据所述人脸训练样本的所述真实人脸检测框和所述人脸检测框预测结果，通过人脸尺寸损失函数计算人脸尺寸损失值；

根据所述人脸训练样本的所述人脸关键点偏置预测结果，以及所述真实人脸检测框的中心点和所述人脸检测框预测结果的中心点的差值，通过人脸关键点偏置损失函数计算人脸关键点偏置损失值；

根据预置权重对所述人脸关键点损失值、所述人脸关键点偏置损失值和所述人脸尺寸损失值进行加权求和，得到最终的网络损失值。

4.一种人脸检测方法，其特征在于，包括：

获取待检测人脸图像；

将所述待检测人脸图像输入到人脸检测模型进行人脸关键点预测和人脸检测框预测，得到所述待检测人脸图像的人脸预测结果，其中，所述人脸检测模型通过权利要求1-3任一项所述的人脸检测模型训练方法得到。

5.一种人脸检测模型训练装置，其特征在于，包括：

构建单元，用于构建人脸检测网络，所述人脸检测网络包括依次连接的主干网络、转置卷积模块和预测模块，所述主干网络中的若干个卷积模块与所述转置卷积模块跳跃连接；

获取单元，用于获取人脸训练样本，所述人脸训练样本的标签信息包括真实人脸关键点坐标和真实人脸检测框；

输入单元，用于将所述人脸训练样本输入到所述人脸检测网络，使得所述主干网络对所述人脸训练样本进行卷积处理，所述转置卷积模块对所述主干网络提取的目标卷积特征图进行上采样处理，得到采样特征图，所述预测模块对所述采样特征图分别进行人脸关键点预测、人脸关键点偏置预测和人脸检测框预测，得到所述人脸训练样本的人脸预测结果，其中，所述目标卷积特征图包括所述主干网络最后一个卷积模块输出的卷积特征图，以及与所述转置卷积模块跳跃连接的卷积模块输出的卷积特征图；

计算单元，用于基于所述人脸训练样本的人脸预测结果和所述标签信息计算网络损失值，并通过所述网络损失值更新所述人脸检测网络的参数，直至所述人脸检测网络收敛，得到人脸检测模型。