基于Transformer的陶瓷軸承表面缺陷檢測方法
組合機床與自動化加工技術(shù)
頁數(shù): 5 2024-02-20
摘要: 針對傳統(tǒng)機器視覺檢測方法中,由于陶瓷軸承滾動體表面曲率大、對比度低,表面成像模糊導(dǎo)致后續(xù)缺陷檢測精度低的問題,提出一種基于Transformer的超分辨率殘差網(wǎng)絡(luò)。首先,網(wǎng)絡(luò)使用殘差學(xué)習(xí)策略,通過預(yù)測模糊圖像與清晰圖像之間的差值,實現(xiàn)超分辨率任務(wù);其次,在網(wǎng)絡(luò)上前端插入通道注意力模塊和空間注意力模塊并改進L2多頭自注意力模塊,以增強圖像紋理、改善梯度爆炸問題;最后,針對超分辨率...