改進(jìn)YOLOv7的晶圓字符檢測(cè)算法
無(wú)線電工程
頁(yè)數(shù): 8 2023-11-02
摘要: 針對(duì)晶圓加工中的字符檢測(cè)問(wèn)題,提出一種基于YOLOv7改進(jìn)的目標(biāo)檢測(cè)模型。在原版YOLOv7的SPP層之前插入Swin Transformer模塊,增強(qiáng)網(wǎng)絡(luò)對(duì)于全局信息的獲取能力,提升對(duì)于全局和局部特征的整合能力;在預(yù)測(cè)部分插入A2-Net注意力機(jī)制,將特征信息全局融合后重新分配,提升網(wǎng)絡(luò)的魯棒性;在定位損失函數(shù)上用SIOU損失函數(shù)代替CIOU,角度損失的引入,增加了對(duì)于字符檢...