摘要:針對目前施工現場(chǎng)的安全帽檢測方法存在遮擋目標檢測難度大、誤檢漏檢率高的問(wèn)題,提出一種改進(jìn)YOLOv5的安全帽檢測方法;首先,使用K-means++聚類(lèi)算法重新設計匹配安全帽數據集的先驗錨框尺寸;其次,使用Swin Transformer作為YOLOv5的骨干網(wǎng)絡(luò )來(lái)提取特征,基于可移位窗口的多頭自注意力機制能建模不同空間位置特征之間的依賴(lài)關(guān)系,有效地捕獲全局上下文信息,具有更好的特征提取能力;再次,提出C3-Ghost模塊,基于Ghost Bottleneck對YOLOv5的C3模塊進(jìn)行改進(jìn), 旨在通過(guò)低成本的操作生成更多有價(jià)值的冗余特征圖,有效減少模型參數和計算復雜度;最后,基于雙向特征金字塔網(wǎng)絡(luò )跨尺度特征融合的結構優(yōu)勢提出新型跨尺度特征融合模塊,更好地適應不同尺度的目標檢測任務(wù);實(shí)驗結果表明,與原始YOLOv5相比,改進(jìn)的YOLOv5在安全帽檢測任務(wù)上的mAP@.5:.95指標提升了2.3%,滿(mǎn)足復雜施工場(chǎng)景下安全帽佩戴檢測的準確率要求。