[發明專利]圖像文本方向分類方法、裝置、設備及存儲介質

在審

• 申請號: 202110232357.2
• 申請日: 2021-03-02
• 公開（公告）號: CN112766418A
• 公開（公告）日: 2021-05-07
• 發明（設計）人: 楊帆;楊兆斌;車成富
• 申請（專利權）人: 陽光財產保險股份有限公司
• 主分類號: G06K9/62
• 分類號: G06K9/62;G06K9/20;G06K9/32
• 代理公司: 北京超凡宏宇專利代理事務所(特殊普通合伙) 11463
• 代理人: 衡滔

1.一種圖像文本方向分類方法，其特征在于，包括：

獲取已標注文本圖像樣本集合；

根據所述已標注文本圖像樣本集合生成文本檢測模型；

根據所述已標注文本圖像樣本集合生成圖像文本方向二分類模型；

將輸入圖像輸入所述文本檢測模型，得到文本字段外接矩形框集合；

從所述文本字段外接矩形框集合中篩選預設數量的外接矩形框，根據所述外接矩形框的寬高比獲取所述輸入圖像中的文本的初步方向；

所述初步方向為橫向或縱向；

根據所述外接矩形框的坐標從所述輸入圖像中裁剪得到對應的預測文本串圖像集合；

將所述預測文本串圖像集合中的圖像輸入所述圖像文本方向二分類模型，得到預測文本串圖像二分類方向集合；

根據所述預測文本串圖像二分類方向集合和所述初步方向獲取所述輸入圖像的文本方向。

2.根據權利要求1所述的圖像文本方向分類方法，其特征在于，

所述已標注文本圖像樣本集合中每張圖像文本方向均為0度，且對應標注信息，所述標注信息包括文本字段外接矩形框位置坐標；

所述根據所述已標注文本圖像樣本集合生成文本檢測模型包括：

將所述已標注文本圖像樣本集合中每張圖像分別依次進行90度，180度，270度旋轉操作，得到新文本圖像集合；

根據所述已標注文本圖像樣本集合中每張圖像的所述標注信息和旋轉角度，計算生成得到轉換后所述新文本圖像集合中每張圖像對應的標注信息；

將所述新文本圖像集合與所述已標注文本圖像樣本集合合并，得到文本圖像集合；

根據所述文本圖像集合生成所述文本檢測模型。

3.根據權利要求2所述的圖像文本方向分類方法，其特征在于，

所述根據所述已標注文本圖像樣本集合生成所述圖像文本方向二分類模型的步驟，包括：

獲取所述已標注文本圖像樣本集合對應的標注信息，根據所述標注信息篩選得到預設數量寬高比大于預設值的文本標注框集合；

根據所述文本標注框集合從所述文本標注框所屬的所述已標注文本圖像樣本集合中的文本圖像對應位置進行裁剪，得到第一訓練文本串圖像樣本集合；

對所述第一訓練文本串圖像樣本集合中每個文本串圖像進行逆時針180度旋轉操作得到第二訓練文本串圖像樣本集合；

根據所述第一訓練文本串圖像樣本集合和所述第二訓練文本串圖像樣本集合，生成所述圖像文本方向二分類模型。

4.根據權利要求3所述的圖像文本方向分類方法，其特征在于，所述根據所述第一訓練文本串圖像樣本集合和所述第二訓練文本串圖像樣本集合，生成所述圖像文本方向二分類模型的具體步驟包括：

將所述第一訓練文本串圖像樣本集合和所述第二訓練文本串圖像樣本集合中每個文本串圖像進行預處理；

將所述第一訓練文本串圖像樣本集合中每張文本串圖像的文本方向標記為0度，第二訓練文本串圖像樣本集合中每張文本串圖像的文本方向標記為180度；

根據標記后的所述第一訓練文本串圖像樣本集合和所述第二訓練文本串圖像樣本集合，通過預設圖像分類算法得到圖像文本方向二分類模型。