[发明专利]一种快速人脸密度预测和人脸检测方法、装置、电子设备及存储介质

权利要求书

1.一种快速人脸密度预测和人脸检测方法，其特征在于，包括以下步骤：

步骤S1：获取待检测图像；

步骤S2：采用特征金字塔残差块提取待检测图像中的多尺度特征；

步骤S3：采用互嵌入上采样模块进行特征融合；

步骤S4：采用人脸检测模块预测人脸置信度和人脸的宽度、高度；

所述互嵌入上采样模块包括：

高阶段特征图上，采用通道注意力模型得到每个通道的第一注意力系数，将所述第一注意力系数和低阶段的特征相乘，得到经过所述通道注意力模型融合的第一融合特征；

低阶段特征图上，采用空间注意力模型得到特征图中每一个点的第二注意力系数，将所述第二注意力系数和经过上采样的所述高阶段特征图相乘，得到经过所述空间注意力模型融合的第二融合特征；

将所述第一融合特征与所述第二融合特征相加，得到最终融合特征。

2.根据权利要求1所述的一种快速人脸密度预测和人脸检测方法，其特征在于，所述步骤S2包括：

步骤S2.1：使用3×3的卷积核与所述待检测图像进行卷积，将卷积后的图像送入到所述特征金字塔残差块中提取特征；

步骤S2.2：使用多个所述特征金字塔残差块组合成一个特征提取网络，提取所述步骤S2.1输出的特征图的特征；

步骤S2.3：使用多个所述特征金字塔残差块组合成一个特征提取网络，提取所述步骤S2.2输出的特征图的特征。

3.根据权利要求2所述的一种快速人脸密度预测和人脸检测方法，其特征在于，所述步骤S3包括：

步骤S3.1：采用所述互嵌入上采样模块将所述步骤S2.2提取的特征与所述步骤S2.3提取的特征进行特征融合；

步骤S3.2：使用所述互嵌入上采样模块将所述步骤S3.1融合后的特征与所述步骤S2.1提取的特征进行特征融合。

4.根据权利要求3所述的一种快速人脸密度预测和人脸检测方法，其特征在于，所述步骤S4包括：

步骤S4.1：使用一个3×3的卷积核卷积所述步骤S3.2融合后的特征；

步骤S4.2：使用两个1×1的卷积核分别预测人脸置信度和人脸的宽度、高度。

5.根据权利要求1所述的一种快速人脸密度预测和人脸检测方法，其特征在于，所述特征金字塔残差块包括：

采用1×1的卷积操作扩展特征图的通道数；

将所述特征图在通道方向平均分成4组，第1组使用空洞大小为1的3×3卷积核卷积所述第1组的特征，第2组使用空洞大小为2的3×3卷积核卷积所述第2组的特征，第3组使用空洞大小为4的3×3卷积核卷积所述第3组的特征，第4组使用空洞大小为8的3×3卷积核卷积所述第4组的特征；

将经卷积核卷积后的4组特征按顺序组合起来组成第一特征图，使用1×1的卷积将所述第一特征图进行特征融合形成第二特征图；

将所述特征图和所述第二特征图相加起来。

6.根据权利要求5所述的一种快速人脸密度预测和人脸检测方法，其特征在于，还包括：

所述第2组在空洞卷积前，将所述第2组特征与所述第1组卷积输出的特征相加；

所述第3组在空洞卷积前，将所述第3组特征与所述第2组卷积输出的特征相加；

所述第4组在空洞卷积前，将所述第4组特征与所述第3组卷积输出的特征相加。

7.根据权利要求6所述的一种快速人脸密度预测和人脸检测方法，其特征在于，还包括：

所述第1组、第2组、第3组、第4组的空洞卷积的感受野分别是3、5、9、17。