小樣本藥用玻璃瓶缺陷生成方法及應(yīng)用

摘要： 目的 為了克服藥包玻璃瓶缺陷樣本不足帶來的缺陷檢測模型精度不準(zhǔn)的問題，提出改進StyleGAN2-ADA的缺陷樣本生成方法，提升模型魯棒性。方法 首先，基于StyleGAN2-ADA算法，在無缺陷圖像集上訓(xùn)練網(wǎng)絡(luò)模型并作為骨干。其次，在骨干網(wǎng)絡(luò)上添加缺陷感知殘差塊，生成缺陷掩碼，在少量的缺陷圖像數(shù)據(jù)集上訓(xùn)練網(wǎng)絡(luò)模型操縱掩碼區(qū)域的特征，模擬缺陷圖像生成過程，合成缺陷圖像。最后，采... （共6頁）