[发明专利]人脸检测方法及相关装置

说明书

本申请实施例公开了一种人脸检测方法及相关装置，应用于电子设备，电子设备中预先配置人脸检测模型，人脸检测模型包括金字塔模型、双注意力模块和多任务损失函数，方法包括：获取目标人脸图像；将目标人脸图像输入到金字塔模型，得到尺度不一样的多个第一特征图；将多个第一特征图中每一特征图输入到双注意力模块中进行运算，得到多个第二特征图，双注意力模块包括空间注意力模块和通道注意力模块；将多个第二特征图进行特征融合，得到中间特征图；将中间特征图输入到多任务损失函数，得到目标人脸检测结果，其中，多任务损失函数包括多个任务，每一任务对应一个任务标签。采用本申请实施例能够提升人脸检测精度。

技术领域

本申请涉及图像处理技术领域，具体涉及一种人脸检测方法及相关装置。

背景技术

人脸检测是所有目标检测子方向中被研究的最充分的问题之一，它在安防监控，人证比对，人机交互，社交和娱乐等方面有很强的应用价值，也是整个人脸识别算法的第一步。人脸检测具体指应用一定的策略对所给的图片或视频来进行检测，判断是否存在着人脸，如果存在则定位出每张人脸的位置、大小与姿态的过程。然而人脸具有复杂的细节变化、存在遮挡以及外在条件如光照、对比度变化等问题，降低了人脸检测精度。

发明内容

本申请实施例提供了一种人脸检测方法及相关装置，可以提升行人脸检测精度。

第一方面，本申请实施例提供一种人脸检测方法，应用于电子设备，所述电子设备中预先配置人脸检测模型，所述人脸检测模型包括金字塔模型、双注意力模块和多任务损失函数，所述方法包括：

获取目标人脸图像；

将所述目标人脸图像输入到所述金字塔模型，得到尺度不一样的多个第一特征图；

将所述多个第一特征图中每一特征图输入到所述双注意力模块中进行运算，得到多个第二特征图，所述双注意力模块包括空间注意力模块和通道注意力模块，每一第二特征图均为添加注意力的特征图；

将所述多个第二特征图进行特征融合，得到中间特征图；

将所述中间特征图输入到所述多任务损失函数，得到目标人脸检测结果，其中，所述多任务损失函数包括多个任务，每一任务对应一个任务标签。

第二方面，本申请实施例提供了一种人脸检测装置，应用于电子设备，所述电子设备中预先配置人脸检测模型，所述人脸检测模型包括金字塔模型、双注意力模块和多任务损失函数，所述装置包括：获取单元、多尺度分解单元、输入单元、融合单元和检测单元，其中，

所述获取单元，用于获取目标人脸图像；

所述多尺度分解单元，用于将所述目标人脸图像输入到所述金字塔模型，得到尺度不一样的多个第一特征图；

所述输入单元，用于将所述多个第一特征图中每一特征图输入到所述双注意力模块中进行运算，得到多个第二特征图，所述双注意力模块包括空间注意力模块和通道注意力模块，每一第二特征图均为添加注意力的特征图；

所述融合单元，用于将所述多个第二特征图进行特征融合，得到中间特征图；

所述检测单元，用于将所述中间特征图输入到所述多任务损失函数，得到目标人脸检测结果，其中，所述多任务损失函数包括多个任务，每一任务对应一个任务标签。

第三方面，本申请实施例提供一种电子设备，包括处理器、存储器、通信接口以及一个或多个程序，其中，上述一个或多个程序被存储在上述存储器中，并且被配置由上述处理器执行，上述程序包括用于执行本申请实施例第一方面中的步骤的指令。

第四方面，本申请实施例提供了一种计算机可读存储介质，其中，上述计算机可读存储介质存储用于电子数据交换的计算机程序，其中，上述计算机程序使得计算机执行如本申请实施例第一方面中所描述的部分或全部步骤。