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

有效

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

1.一種多文檔檢測方法，所述方法包括：

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

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

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

將所述至少一個中心點中的每個中心點與所述多個角點進行匹配，以確定與所述每個中心點關聯的目標角點，其中，通過計算所述每個中心點與每個角點之間的距離，將與中心點距離最小的四個角點確定為與所述中心點關聯的目標角點；

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

2.根據權利要求1所述的多文檔檢測方法，其中，所述從所述中心點預測圖中確定出至少一個中心點，包括：

根據所述中心點預測圖中每個第一特征點的第一預測得分，獲取第一預測得分大于第一得分閾值的第一特征點作為候選中心點；

對所述候選中心點之間的距離小于第一距離閾值的候選中心點進行合并處理，得到所述中心點預測圖中的至少一個中心點。

3.根據權利要求1所述的多文檔檢測方法，其中，所述角點預測圖包括左上角點預測圖、左下角點預測圖、右上角點預測圖和右下角點預測圖；

相應地，所述從所述角點預測圖中確定出多個角點，包括：

根據所述角點預測圖中每個第二特征點的第二預測得分，獲取第二預測得分大于第二得分閾值的第二特征點作為候選角點；

對所述候選角點之間的距離小于第二距離閾值的候選角點進行合并處理，得到所述角點預測圖中的多個角點，所述多個角點包括所述左上角點預測圖中的至少一個左上角點、所述左下角點預測圖中的至少一個左下角點、所述右上角點預測圖中的至少一個右上角點和所述右下角點預測圖中的至少一個右下角點。

4.根據權利要求3所述的多文檔檢測方法，其中，所述目標角點包括目標左上角點、目標左下角點、目標右上角點和目標右下角點。

5.根據權利要求4所述的多文檔檢測方法，其中，所述將所述至少一個中心點中的每個中心點與所述多個角點進行匹配，以確定與所述每個中心點關聯的目標角點，包括：

根據所述每個中心點與所述至少一個左上角點之間的距離，從所述至少一個左上角點中確定出與所述每個中心點距離最近的左上角點，作為與所述每個中心點關聯的目標左上角點；

根據所述每個中心點與所述至少一個左下角點之間的距離，從所述至少一個左下角點中確定出與所述每個中心點距離最近的左下角點，作為與所述每個中心點關聯的目標左下角點；

根據所述每個中心點與所述至少一個右上角點之間的距離，從所述至少一個右上角點中確定出與所述每個中心點距離最近的右上角點，作為與所述每個中心點關聯的目標右上角點；以及

根據所述每個中心點與所述至少一個右下角點之間的距離，從所述至少一個右下角點中確定出與所述每個中心點距離最近的右下角點，作為與所述每個中心點關聯的目標右下角點。

6.根據權利要求5所述的多文檔檢測方法，其中，所述方法還包括：

在與所述每個中心點距離最近的左上角點為多個的情況下，根據與所述每個中心點距離最近的左上角點相對于所述每個中心點的位置，從多個與所述每個中心點距離最近的左上角點中，確定出與所述每個中心點關聯的目標左上角點。