面向小目標(biāo)的改進(jìn)YOLOv5安全帽佩戴檢測(cè)算法
計(jì)算機(jī)工程與應(yīng)用
頁(yè)數(shù): 10 2023-08-17
摘要: 安全帽是施工人員的安全保障,但是現(xiàn)有安全帽檢測(cè)模型在復(fù)雜環(huán)境下對(duì)重疊和密集小目標(biāo)存在誤檢和漏檢等問(wèn)題,為此提出改進(jìn)YOLOv5的小目標(biāo)檢測(cè)算法。在YOLOv5的主干網(wǎng)絡(luò)中加入Transformer捕獲多個(gè)尺度上的全局信息,獲得更豐富的高層語(yǔ)義特征;使用GsConv卷積進(jìn)行特征融合增強(qiáng),并引入坐標(biāo)注意力機(jī)制(coordinate attention),讓網(wǎng)絡(luò)在更大區(qū)域上進(jìn)行注意;...