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