[發明專利]圖像識別方法、裝置、電子設備及存儲介質

有效

• 申請號: 202010124982.0
• 申請日: 2020-02-27
• 公開（公告）號: CN111368893B
• 公開（公告）日: 2023-07-25
• 發明（設計）人: 顏波
• 申請（專利權）人: OPPO廣東移動通信有限公司
• 主分類號: G06V10/774
• 分類號: G06V10/774;G06V10/80;G06V10/764
• 代理公司: 深圳市智圈知識產權代理事務所(普通合伙) 44351
• 代理人: 苗燕

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

獲取多個樣本集，所述樣本集包括多個樣本圖像及所述樣本圖像對應的樣本標簽，其中，同一個樣本集中樣本圖像對應的樣本標簽屬于同一個主類別；

基于初始特征提取模型和所述樣本圖像，得到第一樣本特征信息和第二樣本特征信息，所述初始特征提取模型包括第一特征提取網絡以及第二特征提取網絡，所述第一特征提取網絡用于提取目標圖像的第一特征信息，以及用于作為所述第二特征提取網絡的輸入，并與所述第二特征提取網絡的輸出進行融合；

將所述第一樣本特征信息和所述第二樣本特征信息融合，得到樣本融合特征信息；

根據所述樣本融合特征信息和所述樣本圖像對應的樣本標簽，修正所述初始特征提取模型中所述第二特征提取網絡的網絡參數；

將包含修正后的網絡參數的初始特征提取模型確定為訓練好的特征提取模型；

獲取待識別圖像；

基于所述訓練好的特征提取模型，得到所述待識別圖像的第一特征信息和第二特征信息，其中，所述第一特征信息用于表征所述待識別圖像的目標子類別，所述第二特征信息用于表征所述目標子類別與其他子類別之間的差異，所述目標子類別和所述其他子類別屬于同一個主類別；

將所述第一特征信息和所述第二特征信息進行融合，得到融合特征信息；

根據所述融合特征信息確定所述待識別圖像的識別結果；

根據所述識別結果，執行預定操作。

2.根據權利要求1所述的方法，其特征在于，所述將所述第一樣本特征信息和所述第二樣本特征信息融合，得到樣本融合特征信息，包括：

將所述第一樣本特征信息和所述第二樣本特征信息相加得到所述樣本融合特征信息。

3.根據權利要求1所述的方法，其特征在于，所述基于初始特征提取模型和所述樣本圖像，得到第一樣本特征信息和第二樣本特征信息之前，所述方法還包括：

基于初始第一特征提取網絡，得到所述樣本圖像的第一樣本特征信息；

根據所述第一樣本特征信息和所述樣本圖像對應的樣本標簽，修正所述初始第一特征提取網絡的網絡參數；

將所述初始第一特征提取網絡確定為所述初始特征提取模型的第一特征提取網絡。

4.根據權利要求3所述的方法，其特征在于，所述根據所述第一樣本特征信息和所述樣本圖像對應的樣本標簽，修正所述初始第一特征提取網絡的網絡參數，包括：

根據所述第一樣本特征信息和所述樣本圖像對應的樣本標簽，獲取所述樣本圖像對應的第一損失函數值；

基于所述第一損失函數值修正所述初始第一特征提取網絡的網絡參數。

5.根據權利要求1所述的方法，其特征在于，所述根據所述樣本融合特征信息和所述樣本圖像對應的樣本標簽，修正所述初始特征提取模型中所述第二特征提取網絡的網絡參數，包括：

根據所述樣本融合特征信息和所述樣本圖像對應的樣本標簽，獲取所述樣本圖像對應的第二損失函數值；

基于所述第二損失函數值修正所述第二特征提取網絡的網絡參數。

6.根據權利要求1至5任一項所述的方法，其特征在于，所述第二特征提取網絡包括至少兩層全連接層。

7.根據權利要求1至5任一項所述的方法，其特征在于，所述第一特征提取網絡為MobileNetV2。