[發明專利]物品缺陷檢測方法、裝置、計算機設備及存儲介質

在審

• 申請號: 202211367025.6
• 申請日: 2022-11-01
• 公開（公告）號: CN115690055A
• 公開（公告）日: 2023-02-03
• 發明（設計）人: 周堯;劉樞;呂江波;沈小勇;李志權;王遠
• 申請（專利權）人: 深圳思謀信息科技有限公司
• 主分類號: G06T7/00
• 分類號: G06T7/00;G06V10/764;G06V10/26;G06V10/774;G06V10/82;G06N3/0455;G06N3/0499;G06N3/084
• 代理公司: 北京龍雙利達知識產權代理有限公司 11329
• 代理人: 王君;肖鸝

1.一種物品缺陷檢測方法，其特征在于，包括：

獲取待檢測物品的圖像的第一初始特征集；

基于全局分類器處理所述第一初始特征集，以獲得類別結果和分割結果；

根據所述類別結果和所述分割結果，確定所述待檢測物品的缺陷情況。

2.根據權利要求1所述的物品缺陷檢測方法，其特征在于，所述全局分類器包括訓練完成的編碼器和訓練完成的解碼器，所述基于全局分類器處理第一初始特征集，以獲得類別結果和分割結果，包括：

基于所述訓練完成的編碼器對所述第一初始特征集進行混合編碼，以獲得第一混合特征集；

基于所述訓練完成的解碼器對所述第一混合特征集分別進行類別預測和分割預測，以獲得類別結果和分割結果。

3.根據權利要求2所述的物品缺陷檢測方法，其特征在于，所述第一混合特征集包括M個第一混合特征x_{i_out}；其中，M為大于等于1的正整數，i為序號，0≤i≤M-1，所述基于所述訓練完成的解碼器對所述第一混合特征集分別進行類別預測和分割預測，以獲得類別結果和分割結果，包括：

對第一混合特征x_{M-1-i_out}進行解碼，以獲得第一解碼結果output_i；

對所述第一解碼結果output_i分別進行類別預測和分割預測，以獲得類別結果class_i和分割結果mask_i；其中，所述第一混合特征x_{i_out}的尺寸是按序排列的。

4.根據權利要求3所述的物品缺陷檢測方法，其特征在于，所述對第一混合特征x_{M-1-i_out}進行解碼，以獲得第一解碼結果output_i，包括：

在所述第一混合特征x_{M-1-i_out}的尺寸小于第一混合特征x_{M-i-2_out}的尺寸，和/或，第一混合特征x_{M-i_out}包括第一尺寸、第二尺寸和第三尺寸的混合特征時，基于注意力網絡對所述第一混合特征x_{M-1-i_out}進行解碼，獲得第一解碼結果output_i；或者，

在所述第一混合特征x_{M-1-i_out}的尺寸大于等于第一混合特征x_{M-i-2_out}的尺寸，和/或，第一混合特征x_{M-i_out}包括第一尺寸、第二尺寸和第三尺寸的混合特征時，基于注意力網絡對所述第一混合特征x_{M-1-i_out}和第一解碼結果output_i-1進行解碼，獲得第一解碼結果output_i；其中，所述第一尺寸大于所述第二尺寸，所述第二尺寸大于所述第三尺寸。

5.根據權利要求中3所述的物品缺陷檢測方法，其特征在于，所述對所述第一解碼結果output_i分別進行類別預測和分割預測，以獲得類別結果class_i和分割結果mask_i，包括：

基于預測模塊對所述第一解碼結果output_i和所述第一混合特征x_{M-1-i_out}分別進行類別預測和分割預測，以獲得類別結果class_i和分割結果mask_i。

6.根據權利要求5所述的物品缺陷檢測方法，其特征在于，所述根據所述類別結果和所述分割結果，確定所述待檢測物品的缺陷情況，包括：

若在獲得所述類別結果class_i和所述分割結果mask_i后不再進行解碼處理，則根據所述類別結果class_i和所述分割結果mask_i的乘積，確定所述待檢測物品的缺陷情況。

7.根據權利要求4所述的物品缺陷檢測方法，其特征在于，所述注意力網絡包括交叉注意力網絡、自注意力網絡和第一前饋神經網絡。