基于改进YOLOv5算法的水利工程安全帽佩戴检测与应用
-
摘要: 在水利工程中,佩戴安全帽是预防高处坠落、机械及物体撞击伤害的最经济且有效的防护措施之一。检查作业人员是否佩戴安全帽,是项目安全管理工作的关键环节。针对当前人工检查方法效率低下以及传统检测技术在水利项目特定场景下对小目标安全帽检测准确度不足等问题,提出了一种基于改进YOLOv5算法的识别与检测方法。该方法引入了SIoU_Loss损失函数以替代YOLOv5原有的CIoU_Loss损失函数,旨在提升模型对安全帽的检测精度及快速响应能力。试验结果表明:改进后的YOLOv5模型在安全帽检测任务上取得了0.961的高精度和0.983的高召回率,能够有效满足安全帽的识别与检测需求。
下载: