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

有效

• 申請號: 202110611214.2
• 申請日: 2021-06-01
• 公開（公告）號: CN113239878B
• 公開（公告）日: 2023-09-05
• 發明（設計）人: 周細文;莊伯金;肖京
• 申請（專利權）人: 平安科技（深圳）有限公司
• 主分類號: G06V40/16
• 分類號: G06V40/16;G06V10/40;G06V10/764
• 代理公司: 深圳眾鼎專利商標代理事務所(普通合伙) 44325
• 代理人: 譚果林

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

接收包含待分類圖像的圖像分類指令；

將所述待分類圖像輸入至圖像分類模型中，以通過所述圖像分類模型對所述待分類圖像進行圖像分類，得到圖像分類結果；所述圖像分類模型是根據圖像分類模型訓練方法訓練得到的；所述圖像分類模型訓練方法包括下述步驟：

獲取預設圖像樣本集；所述預設圖像樣本集中包括至少一個樣本分類圖像組；一個所述樣本分類圖像組包括初始正常圖像以及初始缺陷圖像；

將所述初始正常圖像以及所述初始缺陷圖像輸入至包含初始參數的預設分類模型中，以通過所述預設分類模型對所述初始正常圖像以及所述初始缺陷圖像進行解耦恢復處理，確定在解耦恢復處理過程中預設分類模型的圖像恢復損失值、圖像解耦損失值、圖像分類損失值以及特征區分損失值；

根據所述圖像恢復損失值、圖像解耦損失值、圖像分類損失值以及特征區分損失值，確定所述預設分類模型的總損失值；

在所述總損失值未達到預設的收斂條件時，更新迭代所述預設分類模型的初始參數，直至所述總損失值達到所述預設的收斂條件時，將收斂之后的所述預設分類模型記錄為所述圖像分類模型。

2.如權利要求1所述的圖像分類方法，其特征在于，所述通過所述預設分類模型對所述初始正常圖像以及所述初始缺陷圖像進行解耦恢復處理，包括：

對所述初始正常圖像進行圖像解耦處理，得到第一基礎圖像以及第一特征圖像；同時，對所述初始缺陷圖像進行圖像解耦處理，得到第二基礎圖像以及第二特征圖像；

對所述第一基礎圖像以及所述第一特征圖像進行圖像恢復處理，得到恢復正常圖像；對所述第二基礎圖像以及所述第二特征圖像進行圖像恢復處理，得到恢復缺陷圖像；

根據所述第一基礎圖像以及所述第二特征圖像，生成變換缺陷圖像；根據所述第一特征圖像以及所述第二基礎圖像，生成變換正常圖像；

對所述變換正常圖像進行圖像解耦處理，得到第三基礎圖像以及第三特征圖像；對所述變換缺陷圖像進行圖像解耦處理，得到第四基礎圖像以及第四特征圖像。

3.如權利要求2所述的圖像分類方法，其特征在于，所述確定在解耦恢復處理過程中預設分類模型的圖像恢復損失值，包括：

根據所述初始正常圖像以及所述恢復正常圖像，確定正常恢復損失值；同時根據所述初始缺陷圖像以及所述恢復缺陷圖像，確定缺陷恢復損失值；

根據所述恢復損失值以及所述缺陷恢復損失值，確定所述圖像恢復損失值。

4.如權利要求2所述的圖像分類方法，其特征在于，所述確定在解耦恢復處理過程中預設分類模型的圖像解耦損失值，包括：

根據所述第一基礎圖像以及所述第四基礎圖像，確定第一基礎解耦損失值；同時根據所述第二基礎圖像以及所述第三基礎圖像，確定第二基礎解耦損失值；

根據所述第一特征圖像以及所述第三特征圖像，確定第一病征解耦損失值；根據所述第二特征圖像以及所述第四特征圖像，確定第二病征解耦損失值；

將所述第一基礎解耦損失值、第二基礎解耦損失值、第一病征解耦損失值以及第二病征解耦損失值之和記錄為所述圖像解耦損失值。

5.如權利要求2所述的圖像分類方法，其特征在于，所述確定在解耦恢復處理過程中預設分類模型的圖像分類損失值，包括：

確定對所述初始正常圖像以及初始缺陷圖像進行圖像分類的第一分類損失值；

確定對所述恢復正常圖像以及所述恢復缺陷圖像進行圖像分類的第二分類損失值；

確定對所述變換正常圖像以及所述變換缺陷圖像進行圖像分類的第三分類損失值；

將所述第一分類損失值、第二分類損失值以及第三分類損失值之和記錄為所述圖像分類損失值。