[發明專利]圖像分類方法、裝置及存儲介質

在審

• 申請號: 202010432868.4
• 申請日: 2020-05-20
• 公開（公告）號: CN111783822A
• 公開（公告）日: 2020-10-16
• 發明（設計）人: 申世偉
• 申請（專利權）人: 北京達佳互聯信息技術有限公司
• 主分類號: G06K9/62
• 分類號: G06K9/62
• 代理公司: 北京潤澤恒知識產權代理有限公司 11319
• 代理人: 李娜

1.一種圖像分類方法，其特征在于，包括：

獲取原始圖像；

獲取所述原始圖像的特征圖，并根據所述特征圖重建得到與所述原始圖像對應的目標圖像；

根據所述目標圖像，確定所述原始圖像包括的目標的類別信息。

2.根據權利要求1所述的方法，其特征在于，所述獲取所述原始圖像的特征圖，并根據所述特征圖重建得到與所述原始圖像對應的目標圖像，包括：

將所述原始圖像輸入第一目標模型，以通過所述第一目標模型對所述原始圖像進行下采樣，得到所述特征圖，并對所述特征圖進行上采樣，重建得到所述目標圖像，其中，所述第一目標模型為根據第一樣本圖像集對第一預設模型進行訓練獲得的。

3.根據權利要求1或2所述的方法，其特征在于，所述根據所述目標圖像，確定所述原始圖像包括的目標的類別信息，包括：

將所述目標圖像輸入第二目標模型，以通過所述第二目標模型對所述目標圖像進行下采樣，得到所述目標圖像的特征圖，并根據所述目標圖像的特征圖，確定所述類別信息，其中，所述第二目標模型為根據第二樣本圖像集對第二預設模型進行訓練獲得的。

4.根據權利要求1所述的方法，其特征在于，在所述獲取原始圖像之前，還包括：

獲取多個樣本圖像；

將第一樣本圖像輸入第一預設模型，得到與所述第一樣本圖像對應的重建圖像，所述第一樣本圖像為所述多個樣本圖像中的任意一個圖像；

根據所述第一樣本圖像與所述重建圖像對所述第一預設模型進行訓練，得到第一目標模型；

將第二樣本圖像輸入第二預設模型，得到所述第二樣本圖像包括的目標的類別信息，所述第二樣本圖像為所述多個樣本圖像中的任意一個圖像；

根據所述第二樣本圖像的標簽信息和所述第二樣本圖像包括的目標的類別信息對所述第二預設模型進行訓練，得到第二目標模型；

所述獲取所述原始圖像的特征圖，并根據所述特征圖重建得到與所述原始圖像對應的目標圖像，包括：

將所述原始圖像輸入所述第一目標模型，得到所述目標圖像；

所述根據所述目標圖像，確定所述原始圖像包括的目標的類別信息，包括：

將所述目標圖像輸入所述第二目標模型，得到所述類別信息。

5.根據權利要求1所述的方法，其特征在于，在所述獲取原始圖像之前，還包括：

獲取多個樣本圖像；

將目標樣本圖像輸入第一預設模型，得到與所述目標樣本圖像對應的重建圖像，所述目標樣本圖像為所述多個樣本圖像中的任意一個圖像；

將所述重建圖像輸入第二預設模型，得到所述目標樣本圖像包括的目標的類別信息；

根據所述目標樣本圖像的標簽信息和所述目標樣本圖像包括的目標的類別信息對所述第一預設模型和所述第二預設模型進行訓練，得到第一目標模型和第二目標模型；

所述獲取所述原始圖像的特征圖，并根據所述特征圖重建得到與所述原始圖像對應的目標圖像，包括：

將所述原始圖像輸入所述第一目標模型，得到所述目標圖像；

所述根據所述目標圖像，確定所述原始圖像包括的目標的類別信息，包括：

將所述目標圖像輸入所述第二目標模型，得到所述類別信息。

6.一種圖像分類裝置，其特征在于，包括：

第一獲取模塊，被配置為獲取原始圖像；

第二獲取模塊，被配置為獲取所述原始圖像的特征圖，并根據所述特征圖重建得到與所述原始圖像對應的目標圖像；

確定模塊，被配置為根據所述目標圖像，確定所述原始圖像包括的目標的類別信息。

7.根據權利要求6所述的裝置，其特征在于，所述第二獲取模塊具體被配置為將所述原始圖像輸入第一目標模型，以通過所述第一目標模型對所述原始圖像進行下采樣，得到所述特征圖，并對所述特征圖進行上采樣，重建得到所述目標圖像，其中，所述第一目標模型為根據第一樣本圖像集對第一預設模型進行訓練獲得的。