[發明專利]頁面元素的識別方法、裝置、電子設備及存儲介質

在審

• 申請號: 202110442292.4
• 申請日: 2021-04-23
• 公開（公告）號: CN113128604A
• 公開（公告）日: 2021-07-16
• 發明（設計）人: 王繹超
• 申請（專利權）人: 北京奇藝世紀科技有限公司
• 主分類號: G06K9/62
• 分類號: G06K9/62;G06T7/73;G06K9/46;G06N3/04;G06N3/08;G06T7/11;G06T7/136;G06T7/194
• 代理公司: 北京柏杉松知識產權代理事務所(普通合伙) 11413
• 代理人: 項京;丁蕓

1.一種頁面元素的識別方法，其特征在于，所述方法包括：

獲取待處理頁面，所述待處理頁面為包含至少一個目標元素的頁面圖；

基于預先訓練好的圖像識別模型，對所述待處理頁面進行元素識別，獲得識別結果，所述識別結果中包含目標元素的位置及類別。

2.根據權利要求1所述的方法，其特征在于，所述識別結果中還包含：所述目標元素的尺寸信息；

所述方法還包括：

基于各所述目標元素的位置、類別及所述尺寸信息，生成頁面模板結構，所述頁面模板結構中包括各所述目標元素，以及各所述目標元素的排列順序；

基于所述頁面模板結構，以及預設的數據標識，生成頁面模板，所述數據標識為預先確定的用于生成頁面的數據的標識。

3.根據權利要求1所述的方法，其特征在于，所述基于預先訓練好的圖像識別模型，對所述待處理頁面進行元素識別，獲得識別結果的步驟，包括：

利用所述圖像識別模型中的特征提取子模型，對所述待處理頁面進行特征提取，獲得特征圖；

利用所述圖像識別模型中的區域生成子模型，獲得所述特征圖中的多個預測區域，及各所述預測區域對應的目標得分；

利用所述圖像識別模型中的區域篩選子模型，各所述預測區域的目標得分，及各所述預測區域在所述待處理頁面中的位置，從所述多個預測區域篩選出多個目標區域，所述目標區域的數量少于所述預測區域的數量；

利用所述圖像識別模型中的區域識別子模型，以及所述特征圖，獲得各所述目標區域的類別。

4.根據權利要求3所述的方法，其特征在于，所述利用所述圖像識別模型中的區域生成子模型，獲得所述特征圖中的多個預測區域，及各所述預測區域對應的目標得分的步驟，包括：

利用所述區域生成子模型，確定所述特征圖中的多個錨點中心，及每個所述錨點中心對應的多個錨點區域；

獲得每個所述錨點區域對應的預測偏移量及各所述預測偏移量對應的目標得分；所述預測偏移量表示所述錨點區域相對于所述目標元素所預測的偏移量；

利用每個所述錨點區域，及每個所述錨點區域對應的所述多個預測偏移量，獲得每個所述錨點區域對應的多個預測區域。

5.根據權利要求3所述的方法，其特征在于，所述利用所述圖像識別模型中的區域篩選子模型，各所述預測區域的目標得分，及各所述預測區域在所述待處理頁面中的位置，從所述多個預測區域篩選出多個目標區域的步驟，包括：

將各所述預測區域映射至所述待處理頁面中；

將超出所述待處理頁面邊緣的所述預測區域刪除；

針對每個錨點中心對應的經刪除后剩余的預測區域，將最大目標得分對應的預測區域確定為該錨點中心對應的目標區域。

6.根據權利要求3所述的方法，其特征在于，所述利用所述圖像識別模型中的區域識別子模型，以及所述特征圖，獲得各所述目標區域的類別的步驟，包括：

將各所述目標區域映射至所述特征圖，獲得各所述目標區域對應的區域特征圖；

針對各所述區域特征圖，將該區域特征圖劃分為多個子區域特征圖，并對所述多個子區域特征圖中的特征點進行整合處理，獲得經整合后的區域特征圖；

對各所述經整合后的區域特征圖進行識別，獲得各所述目標區域對應的類別。

7.一種頁面元素的識別裝置，其特征在于，所述裝置包括：

頁面獲取模塊，用于獲取待處理頁面，所述待處理頁面為包含至少一個目標元素的頁面圖；

識別結果獲得模塊，用于基于預先訓練好的圖像識別模型，對所述待處理頁面進行元素識別，獲得識別結果，所述識別結果中包含：目標元素的位置及類別。

8.一種電子設備，其特征在于，包括處理器、通信接口、存儲器和通信總線，其中，處理器，通信接口，存儲器通過通信總線完成相互間的通信；

存儲器，用于存放計算機程序；

處理器，用于執行存儲器上所存放的程序時，實現權利要求1-6任一所述的方法步驟。

9.一種計算機可讀存儲介質，其特征在于，所述計算機可讀存儲介質內存儲有計算機程序，所述計算機程序被處理器執行時實現權利要求1-6任一所述的方法步驟。