[發明專利]文本檢測方法、裝置、電子設備和存儲介質

有效

• 申請號: 202210994030.3
• 申請日: 2022-08-18
• 公開（公告）號: CN115393868B
• 公開（公告）日: 2023-05-26
• 發明（設計）人: 牛太陽;王佩雅
• 申請（專利權）人: 中化現代農業有限公司
• 主分類號: G06V30/19
• 分類號: G06V30/19;G06V30/18;G06N3/0464;G06N3/08;G06V10/82
• 代理公司: 北京路浩知識產權代理有限公司 11002
• 代理人: 苗青盛

1.一種文本檢測方法，其特征在于，包括：

將待檢測圖像的第一特征圖輸入至文本檢測模型的注意力層，得到所述注意力層輸出的第二特征圖；

將所述第二特征圖輸入至所述文本檢測模型的文本檢測層，得到所述文本檢測層輸出的文本檢測結果；

其中，所述注意力層用于對所述第一特征圖的多個通道特征進行加權融合；

所述第二特征圖包括多個不同尺度大小的特征圖，所述將所述第二特征圖輸入至所述文本檢測模型的文本檢測層，得到所述文本檢測層輸出的文本檢測結果，包括：

將所述第二特征圖的各特征圖輸入至所述文本檢測層的特征金字塔增強層，得到所述特征金字塔增強層輸出的各增強特征圖；

將所述各增強特征圖輸入至所述文本檢測層的第二特征融合層，得到所述第二特征融合層輸出的融合特征；

將所述融合特征輸入至所述文本檢測層的檢測層，得到所述檢測層輸出的文本檢測結果；

所述特征金字塔增強層包括多個殘差層，所述多個殘差層用于對所述第二特征圖的各特征圖分別進行特征提取；所述殘差層基于如下方式進行特征提取：

將所述第二特征圖的任一特征圖輸入至所述殘差層的第一卷積層和第二卷積層，得到所述第一卷積層輸出的第一子特征圖和所述第二卷積層輸出的第二子特征圖；

將所述第一子特征圖、所述第二子特征圖和所述第二特征圖輸入至所述殘差層的第三特征融合層，得到所述第三特征融合層輸出的特征圖；

所述第二特征圖包括4個特征圖，所述第二特征圖包括第一尺度特征圖、第二尺度特征圖、第三尺度特征圖、第四尺度特征圖，所述殘差層用于對特征圖進行殘差處理，所述將所述第二特征圖的各特征圖輸入至所述文本檢測層的特征金字塔增強層，得到所述特征金字塔增強層輸出的各增強特征圖，包括：依次上采樣和依次下采樣；

所述依次上采樣包括：

將所述第四尺度特征圖與經過殘差處理的第三尺度特征圖進行聚合處理，得到第五尺度特征圖，所述第五尺度特征圖的尺度大小與所述第三尺度特征圖的尺度大小一致；

將所述第五尺度特征圖與經過殘差處理的第二尺度特征圖進行聚合處理，得到第六尺度特征圖，所述第六尺度特征圖的尺度大小與所述第二尺度特征圖的尺度大小一致；

將所述第六尺度特征圖與經過殘差處理的第一尺度特征圖進行聚合處理，得到第七尺度特征圖，所述第七尺度特征圖的尺度大小與所述第一尺度特征圖的尺度大小一致；

所述依次下采樣包括：

將所述第七尺度特征圖與經過殘差處理的第六尺度特征圖進行聚合處理，得到第八尺度特征圖，所述第八尺度特征圖的尺度大小與所述第二尺度特征圖的尺度大小一致；

將所述第八尺度特征圖與經過殘差處理的第五尺度特征圖進行聚合處理，得到第九尺度特征圖，所述第九尺度特征圖的尺度大小與所述第三尺度特征圖的尺度大小一致；

將所述第九尺度特征圖與經過殘差處理的第四尺度特征圖進行聚合處理，得到第十尺度特征圖，所述第十尺度特征圖的尺度大小與所述第四尺度特征圖的尺度大小一致；

也就是說，所述將所述第二特征圖的各特征圖輸入至所述文本檢測層的特征金字塔增強層，得到所述特征金字塔增強層輸出的各增強特征圖，包括：

將所述第一尺度特征圖、所述第二尺度特征圖、所述第三尺度特征圖、所述第四尺度特征圖輸入至所述特征金字塔增強層，得到所述特征金字塔增強層輸出的經過殘差處理的第七尺度特征圖、經過殘差處理的第八尺度特征圖、經過殘差處理的第九尺度特征圖和第十特征圖。

2.根據權利要求1所述的文本檢測方法，其特征在于，所述將待檢測圖像的第一特征圖輸入至文本檢測模型的注意力層，得到所述注意力層輸出的第二特征圖，包括：

將所述第一特征圖輸入至所述注意力層的多通道特征提取層，得到所述多通道特征提取層輸出的至少兩個通道特征；

將所述至少兩個通道特征輸入至所述注意力層的特征加權層，得到所述特征加權層輸出的至少兩個加權特征；

將所述至少兩個加權特征輸入至所述注意力層的第一特征融合層，得到所述第一特征融合層輸出的第二特征圖。