基于改进YOLOv5s的安全帽佩戴检测算法
作者机构:大连交通大学辽宁大连116028
出 版 物:《科技与创新》 (Science and Technology & Innovation)
年 卷 期:2025年第1期
页 面:1-4,9页
学科分类:12[管理学] 1201[管理学-管理科学与工程(可授管理学、工学学位)] 081104[工学-模式识别与智能系统] 08[工学] 080203[工学-机械设计及理论] 0835[工学-软件工程] 0802[工学-机械工程] 0811[工学-控制科学与工程] 0812[工学-计算机科学与技术(可授工学、理学学位)]
基 金:辽宁省自然科学基金计划项目“不确定半马尔科夫跳正时滞系统的鲁棒镇定及其应用研究”(编号:2022-MS-341)
主 题:安全帽佩戴检测 改进YOLOv5s 注意力机制 深度学习
摘 要:在施工中佩戴安全帽是高危行业的最佳防护措施之一。为了解决目前安全帽佩戴检测算法中小目标和密集目标检测困难的问题,对YOLOv5s检测算法进行了改进,提出了YOLOv5-GD检测算法用于安全帽佩戴检测。首先,为了提高YOLOv5s模型的检测精度,用FasterNet网络代替了其骨干网络,以提高计算效率和参数效率;其次,在YOLOv5s的骨干网络中添加了坐标注意力机制,将定位信息嵌入信道注意力中,使网络关注范围更广;再次,增加了小目标检测层,更好地理解多层次特征,可以使网络更加重视检测小目标,提高检测效果,从而增强模型的识别能力;最后,采用更先进的损失函数进行边界框回归(SIoU),可以提高模型训练过程中预测框的收敛速度和有效性。实验结果表明,YOLOv5-GD在自制安全帽数据集上的平均准确率为94.7%,比原始YOLOv5s算法提高了2.7%,在小物体和密集场景中具有更好的检测效果。