融合坐标注意力与多尺度特征的轻量级安全帽佩戴检测 |
| |
引用本文: | 李忠飞,冯仕咏,郭骏,张云鹤,徐飞翔.融合坐标注意力与多尺度特征的轻量级安全帽佩戴检测[J].工矿自动化,2023(11):151-159. |
| |
作者姓名: | 李忠飞 冯仕咏 郭骏 张云鹤 徐飞翔 |
| |
作者单位: | 2. 中国矿业大学信息与控制工程学院;3. 北京和利时数字技术有限公司 |
| |
基金项目: | 国家重点研发计划项目(2021YFC2902702); |
| |
摘 要: | 针对现有煤矿工人安全帽佩戴检测算法存在检测精度与速度难以取得较好平衡的问题,以YOLOv4模型为基础,提出了一种融合坐标注意力与多尺度的轻量级模型M-YOLO,并将其用于安全帽佩戴检测。该模型使用融入混洗坐标注意力模块的轻量化特征提取网络S-MobileNetV2替换YOLOv4的特征提取网络CSPDarknet53,在减少相关参数量的前提下,有效改善了特征之间的联系;将原有空间金字塔池化结构中的并行连接方式改为串行连接,有效提高了计算效率;对特征融合网络进行改进,引入具有高分辨率、多细节纹理信息的浅层特征,以有效加强对检测目标特征的提取,并将原有Neck结构中的部分卷积修改为深度可分离卷积,在保证检测精度的前提下进一步降低了模型的参数量和计算量。实验结果表明,与YOLOv4模型相比,M-YOLO模型的平均精度均值仅降低了0.84%,但计算量、参数量、模型大小分别减小了74.5%,72.8%,81.6%,检测速度提高了53.4%;相较于其他模型,M-YOLO模型在准确率和实时性方面取得了良好的平衡,满足在智能视频监控终端上嵌入式加载和部署的需求。
|
关 键 词: | 目标检测 安全帽佩戴检测 坐标注意力模块 轻量化 多尺度特征融合 |
|
|