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

在審

• 申請號: 202111057748.1
• 申請日: 2021-09-09
• 公開（公告）號: CN113936128A
• 公開（公告）日: 2022-01-14
• 發明（設計）人: 袁夢軻;劉航;王彤;嚴冬明
• 申請（專利權）人: 中國科學院自動化研究所;東華大學
• 主分類號: G06V10/22
• 分類號: G06V10/22;G06V10/46;G06V10/80;G06K9/62;G06N3/04;G06N3/08
• 代理公司: 北京路浩知識產權代理有限公司 11002
• 代理人: 王治東

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

通過特征金字塔將原始光照圖像轉化為對應的參數矩陣，并基于所述參數矩陣調整所述原始光照圖像的圖像亮度，得到所述原始光照圖像對應的待處理暗光圖像；

基于特征提取網絡對所述原始光照圖像對應的原始暗光圖像，以及所述待處理暗光圖像分別進行多層特征提取，分別得到所述原始暗光圖像對應的原始特征和補充特征；

基于所述補充特征對所述原始特征進行特征補全，得到特征補全的原始暗光圖像；

基于輸出網絡輸出所述特征補全的原始暗光圖像的熱力圖和文本預測框，并基于所述熱力圖和所述文本預測框進行文本檢測。

2.根據權利要求1所述的文本檢測方法，其特征在于，所述基于所述熱力圖和所述文本預測框進行文本檢測的步驟包括：

基于所述熱力圖確定所述特征補全的原始暗光圖像中，各個像素點對應的像素分值；

基于各個所述像素分值確定對應的正樣本區域，并基于所述正樣本區域和所述文本預測框進行文本檢測。

3.根據權利要求1所述的文本檢測方法，其特征在于，所述原始特征包括第一部分原始特征和第二部分原始特征，

所述基于所述補充特征對所述原始特征進行特征補全，得到特征補全的原始暗光圖像的步驟包括：

將所述補充特征和所述第一部分原始特征，輸入至第一特征融合模塊，其中，所述第一特征融合模塊包括第一合并函數和第一卷積層；

基于所述第一合并函數和所述第一卷積層，將所述補充特征補全至所述第一部分原始特征中；

將所述第二部分原始特征及其對應的上一級特征進行特征融合，得到所述特征補全的原始暗光圖像。

4.根據權利要求3所述的文本檢測方法，其特征在于，所述第二部分原始特征包括第一特征和第二特征，所述第一特征在所述第二特征的底層，

所述將所述第二部分原始特征及其對應的上一級特征進行特征融合，得到所述特征補全的原始暗光圖像的步驟包括：

獲取所述第一特征對應的第一上一級特征，將所述第一特征和所述第一上一級特征輸入至第二特征融合模塊，其中，所述第二特征融合模塊包括第二合并函數和第二卷積層；

基于所述第二合并函數和所述第二卷積層，將所述第一特征和所述第一上一級特征進行第一次特征融合；

獲取所述第二特征對應的第二上一級特征，基于所述第二合并函數和所述第二卷積層，將所述第二特征和所述第二上一級特征進行第二次特征融合；

基于所述第一次特征融合和所述第二次特征融合，得到所述特征補全的原始暗光圖像。

5.根據權利要求1所述的文本檢測方法，其特征在于，所述基于所述參數矩陣調整所述原始光照圖像的圖像亮度，得到所述原始光照圖像對應的待處理暗光圖像的步驟包括：

確定所述原始光照圖像中各個圖像點的像素值，并基于所述參數矩陣結合預設迭代函數對各個所述像素值進行迭代調整，得到各個所述像素值對應的調整后的像素值；

基于各個所述調整后的像素值，及平均亮度的約束信息調整所述原始光照圖像的圖像亮度，得到所述待處理暗光圖像。

6.根據權利要求1所述的文本檢測方法，其特征在于，所述基于特征提取網絡對所述原始光照圖像對應的原始暗光圖像進行多層特征提取，得到所述原始暗光圖像對應的原始特征的步驟包括：

基于所述特征提取網絡對原始暗光圖像進行多層特征提取，得到原始暗光圖像對應的預設層特征，并將所述預設層特征確定為所述原始特征。

7.根據權利要求1至6任一項所述的文本檢測方法，其特征在于，所述通過特征金字塔將原始光照圖像轉化為對應的參數矩陣，并基于所述參數矩陣調整所述原始光照圖像的圖像亮度，得到所述原始光照圖像對應的待處理暗光圖像的步驟之前，還包括：

通過攝像設備拍攝在預設正常光照環境下帶有文本數據的圖像，得到對應的原始光照圖像；

通過所述攝像設備拍攝在預設暗光環境下所述原始光照圖像對應的圖像，得到所述原始光照圖像對應的原始暗光圖像。