基于STA-YOLOv5的水利建造人员安全帽佩戴检测算法

摘要：

在大型水利建造工程现场,存在高空坠物、塔吊转动、墙体坍塌等问题,对于建造人员人身安全造成巨大威胁,佩戴安全帽是保护建造人员的有效措施,作为工程作业中的安全管理,对建造人员进行安全帽佩戴的精确检测很有必要.针对现有安全帽检测算法在大型水利建造场景下对小且密集的安全帽目标存在漏检、检测精度较低等问题,提出一种基于STA-YOLOv5的安全帽佩戴检测算法,该算法将Swin Transformer和注意力机制引入到YOLOv5算法中,提高模型对安全帽的识别能力.实验结果表明,STA-YOLOv5算法具有更精确检测结果,识别准确率达到91.6%,较原有的YOLOv5算法有明显提升.

关键词：水利建造、安全帽佩戴检测、STA-YOLOv5、Swin Transformer、注意力机制

所属期刊栏目：46

分类号：TP391.4(计算技术、计算机技术)

在线出版日期：2023-10-12（万方平台首次上网日期，不代表论文的发表时间）

页数：共11页

页码：142-152