[发明专利]文本检测方法、装置、电子设备和存储介质

权利要求书

1.一种文本检测方法，其特征在于，包括：

将待检测图像的第一特征图输入至文本检测模型的注意力层，得到所述注意力层输出的第二特征图；

将所述第二特征图输入至所述文本检测模型的文本检测层，得到所述文本检测层输出的文本检测结果；

其中，所述注意力层用于对所述第一特征图的多个通道特征进行加权融合；

所述第二特征图包括多个不同尺度大小的特征图，所述将所述第二特征图输入至所述文本检测模型的文本检测层，得到所述文本检测层输出的文本检测结果，包括：

将所述第二特征图的各特征图输入至所述文本检测层的特征金字塔增强层，得到所述特征金字塔增强层输出的各增强特征图；

将所述各增强特征图输入至所述文本检测层的第二特征融合层，得到所述第二特征融合层输出的融合特征；

将所述融合特征输入至所述文本检测层的检测层，得到所述检测层输出的文本检测结果；

所述特征金字塔增强层包括多个残差层，所述多个残差层用于对所述第二特征图的各特征图分别进行特征提取；所述残差层基于如下方式进行特征提取：

将所述第二特征图的任一特征图输入至所述残差层的第一卷积层和第二卷积层，得到所述第一卷积层输出的第一子特征图和所述第二卷积层输出的第二子特征图；

将所述第一子特征图、所述第二子特征图和所述第二特征图输入至所述残差层的第三特征融合层，得到所述第三特征融合层输出的特征图；

所述第二特征图包括4个特征图，所述第二特征图包括第一尺度特征图、第二尺度特征图、第三尺度特征图、第四尺度特征图，所述残差层用于对特征图进行残差处理，所述将所述第二特征图的各特征图输入至所述文本检测层的特征金字塔增强层，得到所述特征金字塔增强层输出的各增强特征图，包括：依次上采样和依次下采样；

所述依次上采样包括：

将所述第四尺度特征图与经过残差处理的第三尺度特征图进行聚合处理，得到第五尺度特征图，所述第五尺度特征图的尺度大小与所述第三尺度特征图的尺度大小一致；

将所述第五尺度特征图与经过残差处理的第二尺度特征图进行聚合处理，得到第六尺度特征图，所述第六尺度特征图的尺度大小与所述第二尺度特征图的尺度大小一致；

将所述第六尺度特征图与经过残差处理的第一尺度特征图进行聚合处理，得到第七尺度特征图，所述第七尺度特征图的尺度大小与所述第一尺度特征图的尺度大小一致；

所述依次下采样包括：

将所述第七尺度特征图与经过残差处理的第六尺度特征图进行聚合处理，得到第八尺度特征图，所述第八尺度特征图的尺度大小与所述第二尺度特征图的尺度大小一致；

将所述第八尺度特征图与经过残差处理的第五尺度特征图进行聚合处理，得到第九尺度特征图，所述第九尺度特征图的尺度大小与所述第三尺度特征图的尺度大小一致；

将所述第九尺度特征图与经过残差处理的第四尺度特征图进行聚合处理，得到第十尺度特征图，所述第十尺度特征图的尺度大小与所述第四尺度特征图的尺度大小一致；

也就是说，所述将所述第二特征图的各特征图输入至所述文本检测层的特征金字塔增强层，得到所述特征金字塔增强层输出的各增强特征图，包括：

将所述第一尺度特征图、所述第二尺度特征图、所述第三尺度特征图、所述第四尺度特征图输入至所述特征金字塔增强层，得到所述特征金字塔增强层输出的经过残差处理的第七尺度特征图、经过残差处理的第八尺度特征图、经过残差处理的第九尺度特征图和第十特征图。

2.根据权利要求1所述的文本检测方法，其特征在于，所述将待检测图像的第一特征图输入至文本检测模型的注意力层，得到所述注意力层输出的第二特征图，包括：

将所述第一特征图输入至所述注意力层的多通道特征提取层，得到所述多通道特征提取层输出的至少两个通道特征；

将所述至少两个通道特征输入至所述注意力层的特征加权层，得到所述特征加权层输出的至少两个加权特征；

将所述至少两个加权特征输入至所述注意力层的第一特征融合层，得到所述第一特征融合层输出的第二特征图。