
基于改良版YOLOv5的安全帽佩戴检测算法
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本研究提出了一种改进的YOLOv5算法,专注于提高施工现场安全帽佩戴情况的检测准确率和效率,保障作业人员的安全。
为了应对现有安全帽佩戴检测中存在的干扰性强、精度低等问题,本段落提出了一种基于改进YOLOv5的安全帽检测新算法。首先,针对不同尺寸的安全帽问题,使用K-Means++算法重新设计了先验框的尺寸,并将其匹配到相应的特征层;其次,在特征提取网络中引入多光谱通道注意力模块,使模型能够自主学习每个通道的重要性权重,增强信息传播能力,从而提高对前景和背景的区分效果。此外,在训练过程中通过随机输入不同大小的图像来提升算法在实际场景中的泛化性能。
实验结果显示,该方法在自制的安全帽佩戴检测数据集上取得了显著的效果:均值平均精度(mAP)达到了96.0%,其中对于戴安全帽工人的平均精度(AP)为96.7%,未戴安全帽的工人则为95.2%。与原始YOLOv5算法相比,改进后的模型在检测佩戴安全帽方面的准确率提高了3.4个百分点,从而满足了施工环境中对安全帽佩戴情况精确监测的需求。
全部评论 (0)
还没有任何评论哟~


