基于改進(jìn)YOLOv5的陶瓷表面缺陷檢測算法

打開文本圖片集

摘  要：提出一種陶瓷表面缺陷檢測算法YOLOv5-G。該算法在YOLOv5框架的基礎(chǔ)上，將全局注意力機(jī)制（GAM）引入主干和頸部網(wǎng)絡(luò)中，該機(jī)制能夠在減少信息彌散的情況下放大全局交互特征，增強(qiáng)了網(wǎng)絡(luò)的特征表達(dá)能力；使用α-CIoU作為改進(jìn)算法的邊界框回歸損失函數(shù)，自適應(yīng)地向上加權(quán)高IoU對象的損失和梯度，使得模型可以更加關(guān)注IoU高的目標(biāo)，從而幫助提高定位精度。（剩余11622字）