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

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