基于注意力特征融合的無(wú)人機(jī)檢測(cè)算法

摘要： 針對(duì)背景與無(wú)人機(jī)(UAV)顏色相似、目標(biāo)遮擋重疊等復(fù)雜環(huán)境條件下，對(duì)無(wú)人機(jī)目標(biāo)快速精準(zhǔn)識(shí)別存在的問(wèn)題，提出一種改進(jìn)的無(wú)人機(jī)檢測(cè)算法STC-YOLOv5。STC-YOLOv5算法的骨干特征提取網(wǎng)絡(luò)采用Swin Transformer，以增強(qiáng)網(wǎng)絡(luò)在復(fù)雜環(huán)境下的魯棒性；在YOLOv5模型特征融合網(wǎng)絡(luò)集成卷積注意力模塊(CBAM)，以增強(qiáng)對(duì)無(wú)人機(jī)目標(biāo)重要特征的關(guān)注并抑制不必要的特征關(guān)注... （共8頁(yè)）