• 《工程索引》(EI)刊源期刊
  • 中文核心期刊
  • 中国科技论文统计源期刊
  • 中国科学引文数据库来源期刊

留言板

尊敬的读者、作者、审稿人, 关于本刊的投稿、审稿、编辑和出版的任何问题, 您可以本页添加留言。我们将尽快给您答复。谢谢您的支持!

姓名
邮箱
手机号码
标题
留言内容
验证码

基于改进YOLOv5的安全帽检测算法

侯公羽 陈钦煌 杨振华 张又文 张丹阳 李昊翔

侯公羽, 陈钦煌, 杨振华, 张又文, 张丹阳, 李昊翔. 基于改进YOLOv5的安全帽检测算法[J]. 工程科学学报. doi: 10.13374/j.issn2095-9389.2022.12.07.002
引用本文: 侯公羽, 陈钦煌, 杨振华, 张又文, 张丹阳, 李昊翔. 基于改进YOLOv5的安全帽检测算法[J]. 工程科学学报. doi: 10.13374/j.issn2095-9389.2022.12.07.002
Helmet detection method based on improved YOLOv5[J]. Chinese Journal of Engineering. doi: 10.13374/j.issn2095-9389.2022.12.07.002
Citation: Helmet detection method based on improved YOLOv5[J]. Chinese Journal of Engineering. doi: 10.13374/j.issn2095-9389.2022.12.07.002

基于改进YOLOv5的安全帽检测算法

doi: 10.13374/j.issn2095-9389.2022.12.07.002
详细信息
  • 中图分类号: TP391.4

Helmet detection method based on improved YOLOv5

  • 摘要: 为了解决建筑工地、隧道、煤矿等施工场景中现有安全帽检测算法对于小目标、密集目标以及复杂环境下的检测精度低的问题,提出了一种基于YOLOv5的改进目标检测算法,记为YOLOv5-GBC。首先使用Ghost卷积对骨干网络进行重构,使得模型的复杂度有了显著降低。其次使用双向特征金字塔网络(BiFPN)加强特征融合,使得算法对小目标和密集目标准确率提升。最后引入坐标注意力(Coordinate Attention)模块,能够将注意力资源分配给关键区域,从而在复杂环境中降低背景的干扰。为了验证算法的可行性,以课题组收集的安全帽数据集为基础,选用了多种经典算法进行对比,并且进行了消融实验,探究各个改进模块的提升效果。实验结果表明:改进算法YOLOv5-GBC相较于YOLOv5s算法,算法平均精确率(IOU=0.5)提升了4.9%,达到了93.6%,检测速度达到了124.3FPS,模型更加轻量化,在密集场景和小目标场景下检测能力提升显著,并且同时满足安全帽检测精度和实时性的要求,给复杂施工环境下安全帽检测提供了一种新的方法。

     

  • 加载中
计量
  • 文章访问数:  164
  • HTML全文浏览量:  13
  • PDF下载量:  46
  • 被引次数: 0
出版历程
  • 网络出版日期:  2023-05-05

目录

    /

    返回文章
    返回