摘要
本发明公开了一种复杂环境下的安全帽佩戴检测方法,属于计算机视觉技术领域,本发明的方法包括以下步骤:构建并按比例划分安全帽图像数据集;在主干网络中采用C2f‑MSEFA模块替换C2f模块,并提出了MAConv模块;在颈部网络中引入ASF‑FS结构;在检测头部分,采用EELD检测头。基于上述结构,构建MMAE‑YOLO模型,并通过设定合理的网络参数进行迭代训练,结合实验结果优化超参数配置,最终获得最优的网络模型结构。使用MMAE‑YOLO模型,对待检测的安全帽数据集进行佩戴检测,并输出检测结果。本发明所述的MMAE‑YOLO算法在保证实时性的前提下,较YOLOv8n模型显著提升了对小目标及遮挡目标的检测准确率,并有效降低了模型复杂度,适用于施工现场的高效且可靠的安全监测。
技术关键词
YOLO模型
检测头
跨尺度特征融合
网络模型结构
注意力机制
YOLO算法
特征选择
多尺度
计算机视觉技术
佩戴安全帽
超参数
预测类别
通道
卷积模块
语义特征
施工现场