[發明專利]一種電子文件的旋轉校正方法和裝置

在審

• 申請號: 202011422419.8
• 申請日: 2020-12-08
• 公開（公告）號: CN112364868A
• 公開（公告）日: 2021-02-12
• 發明（設計）人: 趙儀琳;王偉偉;王騰飛
• 申請（專利權）人: 共道網絡科技有限公司
• 主分類號: G06K9/32
• 分類號: G06K9/32;G06K9/62;G06T3/60
• 代理公司: 北京博思佳知識產權代理有限公司 11415
• 代理人: 王茹

1.一種電子文件的旋轉校正方法，其特征在于，所述方法包括：

獲取待校正的目標電子文件圖像；

將所述目標電子文件圖像輸入角度識別模型，所述角度識別模型包括特征提取層、特征融合層、特征轉換層和分類層；

其中，所述特征提取層用于基于若干不同的特征提取方向對所述目標電子文件圖像進行特征提取，得到各個特征提取方向對應的初始特征；

所述特征融合層用于將各個特征提取方向對應的初始特征進行融合，得到綜合特征；

所述特征轉換層用于將所述綜合特征轉換為若干個標準旋轉角度對應的概率值，所述標準旋轉角度的數量與角度識別精度正相關；

所述分類層用于基于各個標準旋轉角度對應的概率值確定所述目標電子文件圖像的校正旋轉角度；

基于所述角度識別模型輸出的校正旋轉角度對所述目標電子文件圖像進行校正。

2.根據權利要求1所述方法，其特征在于，

所述特征提取層包括若干個特征提取模塊，每個特征提取模塊執行所述基于若干不同的特征提取方向對所述目標電子文件圖像進行特征提取，得到與所述特征提取模塊對應的各個特征提取方向的初始特征；

所述特征融合層包括第一子特征融合層、子池化層和第二子特征融合層，其中，所述第一子特征融合層用于針對每個特征提取模塊對應的各個特征提取方向的初始特征，將所述各個特征提取方向的初始特征進行融合，得到所述特征提取模塊對應的中間特征；

所述子池化層用于分別對每個特征提取模塊對應的中間特征進行池化處理，得到每個特征提取模塊對應的池化處理后的中間特征；

所述第二子特征融合層用于對各個特征提取模塊對應的池化處理后的中間特征進行融合，得到所述目標電子文件圖像的綜合特征。

3.根據權利要求1或2所述方法，其特征在于，所述基于若干不同的特征提取方向對所述目標電子文件圖像進行特征提取，得到各個特征提取方向對應的初始特征，包括：

分別采用第一特征提取方向對應的第一卷積核和第二特征提取方向對應的第二卷積核對所述目標電子文件進行特征提取，得到第一特征提取方向對應的初始特征和第二特征提取方向對應的初始特征，其中，所述第一卷積核和所述第二卷積核互為轉置矩陣。

4.根據權利要求1所述方法，其特征在于，特征轉換層將所述綜合特征轉換為若干個標準旋轉角度對應的概率值，包括：

基于reshape函數將所述綜合特征轉換為旋轉角度向量，其中，所述旋轉角度向量中元素的數量與所述標準旋轉角度的數量相同，所述旋轉角度向量的每個元素值分別代表對應標準旋轉角度的概率值。

5.根據權利要求1所述方法，其特征在于，所述基于各個標準旋轉角度對應的概率值，確定所述目標電子文件圖像的校正旋轉角度，包括：

將概率值高的標準旋轉角度確定為所述目標電子文件圖像的校正旋轉角度。

6.根據權利要求1所述方法，其特征在于，所述待校正的目標電子文件圖像通過掃描紙質文件得到。

7.一種電子文件的旋轉校正裝置，其特征在于，所述裝置包括：

獲取單元，用于獲取待校正的目標電子文件圖像；

輸入單元，用于將所述目標電子文件圖像輸入角度識別模型，所述角度識別模型包括特征提取層、特征融合層、特征轉換層和分類層；

其中，所述特征提取層用于基于若干不同的特征提取方向對所述目標電子文件圖像進行特征提取，得到各個特征提取方向對應的初始特征；

所述特征融合層用于將各個特征提取方向對應的初始特征進行融合，得到綜合特征；

所述特征轉換層用于將所述綜合特征轉換為若干個標準旋轉角度對應的概率值，所述標準旋轉角度的數量與角度識別精度正相關；

所述分類層用于基于各個標準旋轉角度對應的概率值確定所述目標電子文件圖像的校正旋轉角度；

校正單元，用于基于所述角度識別模型輸出的校正旋轉角度對所述目標電子文件圖像進行校正。