[發明專利]一種圖像識別的方法及設備

在審

• 申請號: 202210614363.9
• 申請日: 2022-05-31
• 公開（公告）號: CN115019293A
• 公開（公告）日: 2022-09-06
• 發明（設計）人: 朱亞倫;沈孔懷
• 申請（專利權）人: 浙江大華技術股份有限公司
• 主分類號: G06V20/62
• 分類號: G06V20/62;G06V10/56;G06V10/44;G06V10/82;G06V10/764;G06V30/244;G06V30/19;G06V20/70;G06N3/04;G06N3/08
• 代理公司: 北京同達信恒知識產權代理有限公司 11291
• 代理人: 趙凱莉

1.一種圖像識別的方法，其特征在于，該方法包括：

確定待識別的圖像和所述圖像中對象的外部特征；

將所述圖像輸入到訓練好的識別模型中，利用主干網絡提取所述圖像的特征得到特征圖，將所述特征圖分別輸入到第一分支網絡和第二分支網絡，對應輸出第一預測結果和第二預測結果；其中所述識別模型包括所述主干網絡、所述第一分支網絡和所述第二分支網絡；

根據所述對象的外部特征、所述第一預測結果和第二預測結果，確定所述對象的識別結果。

2.根據權利要求1所述的方法，其特征在于，通過如下方式確定所述識別模型訓練完成：

獲取標注的第一訓練集和標注的第二訓練集；

將所述第一訓練集經過主干網絡輸入到第一分支網絡，計算第一損失值，以及將所述第二訓練集經過主干網絡輸入到第二分支網絡，計算第二損失值；

分別利用所述第一損失值和所述第二損失值，對所述識別模型進行交替訓練，當所述第一損失值和所述第二損失值都滿足預設條件時，確定所述識別模型訓練完成。

3.根據權利要求2所述的方法，其特征在于，通過如下方式確定所述第二訓練集：

獲取標注的真實圖像，利用生成的至少一個第一字符，對所述真實圖像中的至少一個第二字符進行替換，得到合成圖像；

根據進行替換的至少一個第一字符標注的標簽，對所述合成圖像進行標注；

根據標注的合成圖像，確定第二訓練集。

4.根據權利要求1所述的方法，其特征在于，所述根據所述對象的外部特征、所述第一預測結果和第二預測結果，確定所述對象的識別結果，包括：

當所述對象的外部特征滿足第一預設特征時，根據所述第一預測結果確定所述對象的識別結果，其中所述第一預設特征是根據所述第一訓練集的圖像中對象的外部特征確定的；或，

當所述對象的外部特征滿足第二預設特征時，根據所述第二預測結果確定所述對象的識別結果，其中所述第二預設特征是根據所述第二訓練集的圖像中對象的外部特征確定的。

5.根據權利要求4所述的方法，其特征在于，所述當所述對象的外部特征滿足第一預設特征時，根據所述第一預測結果確定所述對象的識別結果，包括：

確定與所述第一預設特征對應的第一字符規則；

當所述第一預測結果符合所述第一字符規則時，根據所述第一預測結果確定所述對象的識別結果。

6.根據權利要求4所述的方法，其特征在于，所述當所述對象的外部特征滿足第二預設特征時，根據所述第二預測結果確定所述對象的識別結果，包括：

確定與所述第二預設特征對應的第二字符規則；

當所述第二預測結果符合所述第二字符規則時，根據所述第二預測結果確定所述對象的識別結果。

7.根據權利要求6所述的方法，其特征在于，當所述第二預測結果不符合所述第二字符規則時，還包括：

確定與所述第一預設特征對應的第一字符規則；

當所述第二預測結果符合所述第一字符規則時，根據所述第二預測結果確定所述對象的識別結果。

8.根據權利要求1～7任一所述的方法，其特征在于，

所述第一預測結果包括至少一個置信度大于第一閾值的第一預測標簽；和/或，

所述第二預測結果包括至少一個置信度大于第二閾值的第二預測標簽。

9.一種圖像識別的設備，其特征在于，該設備包括處理器和存儲器，所述存儲器用于存儲所述處理器可執行的程序，所述處理器用于讀取所述存儲器中的程序并執行權利要求1～8任一所述方法的步驟。

10.一種計算機存儲介質，其上存儲有計算機程序，其特征在于，該程序被處理器執行時實現如權利要求1～8任一所述方法的步驟。