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

有效

• 申請號: 202111433043.5
• 申請日: 2021-11-29
• 公開（公告）號: CN113850805B
• 公開（公告）日: 2022-02-22
• 發明（設計）人: 張子浩
• 申請（專利權）人: 北京世紀好未來教育科技有限公司
• 主分類號: G06T7/00
• 分類號: G06T7/00;G06T7/13;G06V30/19;G06K9/62
• 代理公司: 北京開陽星知識產權代理有限公司 11710
• 代理人: 王艷斌

本公開提供一種多文檔檢測方法、裝置、電子設備及存儲介質，所述方法包括：獲取包含至少一個文檔的待檢測圖像，將待檢測圖像輸入至預先訓練的中心點和角點檢測模型，以獲取待檢測圖像的中心點預測圖和角點預測圖，接著，從中心點預測圖中確定出至少一個中心點，以及從角點預測圖中確定出多個角點，并將至少一個中心點中的每個中心點與多個角點進行匹配，以確定與每個中心點關聯的目標角點，進而基于與每個中心點關聯的目標角點，采用透視變換對待檢測圖像進行裁剪，得到至少一個文檔中的每個文檔對應的文檔圖像，由此，可以實現對圖像中包含的至少一個文檔進行檢測，實現對圖像中的多個文檔進行拆分。

技術領域

本公開涉及深度學習技術領域，尤其涉及一種多文檔檢測方法、裝置、電子設備及存儲介質。

背景技術

隨著計算機技術和人工智能技術的不斷發展，人工智能技術已逐步應用在教育教學場景中，比如拍照搜題、智能批改、題目錄入等。

在教育教學場景中，學生或教師在拍攝試卷、作業等的圖像時，可能會對多張文檔頁面同時拍攝，使得一張圖像中包含多張文檔。然而，現有的文檔檢測技術，通常僅能檢測出圖像中的單頁文檔，而不支持多頁文檔檢測，從而使得對包含多個文檔的圖像進行批改的準確率較低。

因此，如何將圖像中的多個文檔進行拆分成為亟待解決的問題。

發明內容

為了解決上述技術問題或者至少部分地解決上述技術問題，本公開實施例提供了一種多文檔檢測方法、裝置、電子設備及存儲介質。

根據本公開的一方面，提供了一種多文檔檢測方法，包括：

獲取待檢測圖像，所述待檢測圖像中包含至少一個文檔；

將所述待檢測圖像輸入至預先訓練的中心點和角點檢測模型，以獲取所述待檢測圖像的中心點預測圖和角點預測圖；

從所述中心點預測圖中確定出至少一個中心點，以及，從所述角點預測圖中確定出多個角點；

將所述至少一個中心點中的每個中心點與所述多個角點進行匹配，以確定與所述每個中心點關聯的目標角點；

基于與所述每個中心點關聯的目標角點，采用透視變換對所述待檢測圖像進行裁剪，得到所述至少一個文檔中的每個文檔對應的文檔圖像。

根據本公開的另一方面，提供了一種多文檔檢測裝置，包括：

圖像獲取模塊，用于獲取待檢測圖像，所述待檢測圖像中包含至少一個文檔；

預測圖獲取模塊，用于將所述待檢測圖像輸入至預先訓練的中心點和角點檢測模型，以獲取所述待檢測圖像的中心點預測圖和角點預測圖；

確定模塊，用于從所述中心點預測圖中確定出至少一個中心點，以及，從所述角點預測圖中確定出多個角點；

匹配模塊，用于將所述至少一個中心點中的每個中心點與所述多個角點進行匹配，以確定與所述每個中心點關聯的目標角點；

裁剪模塊，用于基于與所述每個中心點關聯的目標角點，采用透視變換對所述待檢測圖像進行裁剪，得到所述至少一個文檔中的每個文檔對應的文檔圖像。

根據本公開的另一方面，提供了一種電子設備，包括：

處理器；以及

存儲程序的存儲器，

其中，所述程序包括指令，所述指令在由所述處理器執行時使所述處理器執行根據前述一方面所述的多文檔檢測方法。

根據本公開的另一方面，提供了一種存儲有計算機指令的非瞬時計算機可讀存儲介質，其中，所述計算機指令用于使所述計算機執行根據前述一方面所述的多文檔檢測方法。