火电厂监控视频安全帽检测方法研究

作者：游锦伟; 薛涛; 于俊清^*

来源：湖北电力, 2022, 46(02): 97-103.

DOI：10.19308/j.hep.2022.02.015

• 目标检测
• 安全帽检测
• YOLOv5
• YOLO
• 工业视频监控

安全帽是火电厂生产环境中非常重要的安全防护装备，利用计算机视觉领域中的相关技术对其检测能有效保障工业生产安全。针对安全帽在复杂工业环境中易受到光照环境变化、多遮挡、目标过小导致检测困难的问题，构建了安全帽目标检测数据集，改进了YOLOv5目标检测算法的Head模块，对Anchor的数目与大小进行了调整，通过k-means聚类算法对安全帽数据集重新聚类，增加自顶向下网络特征提取过程中的上采样模块，对提取到的特征图进一步扩大，并将深层特征与浅层特征信息充分融合，增强了模型多尺度特征提取能力。基于构建的工业场景安全帽目标检测数据集，将改进后的YOLOv5算法与目前相关领域最优算法进行对比实验。实验结果表明，改进后的YOLOv5算法在准确率方面均得到显著提升，最高检测率达到了98.1%。综合考虑算法检测速度、精度与模型体积等多种因素，改进后的YOLOv5算法可以满足工业场景中检测的需求。

• 单位
  华中科技大学