• 论文
主办单位:煤炭科学研究总院有限公司、中国煤炭学会学术期刊工作委员会
基于改进YOLOv7的矿用电铲检测算法
  • 72
  • 作者

    宋立业赵小萱崔昊

  • 单位

    辽宁工程技术大学电气与控制工程学院

  • 摘要
    针对现有基于深度学习的电铲检测方法未能很好地平衡检测速度与检测精度的问题,提出了一种改进YOLOv7模型,并将其用于矿用电铲检测。该模型以YOLOv7模型为基础,在主干网络中采用轻量化GhostNet网络进行特征提取,在颈部网络中采用轻量级GSConv替换部分普通卷积,以减少模型参数量和计算量,提高模型检测速度;考虑到轻量化改进后模型参数量减少对特征信息提取能力的影响,在不增加计算量的前提下,对颈部网络进行进一步改进,在扩展高效层聚合网络(ELAN)中嵌入坐标注意力机制(CA),同时利用双向特征金字塔网络(BiFPN)改进路径聚合网络(PANet),以提高网络对特征信息的提取能力,进而有效提高模型检测精度。实验结果表明,与YOLOv7模型相比,改进YOLOv7模型的参数量减少了75.4%,每秒浮点运算次数减少了82.9%,检测速度提高了24.3%;相较于其他目标检测模型,改进YOLOv7模型在检测速度和检测精度方面取得了良好的平衡,满足在露天煤矿场景下对电铲进行实时、准确检测的需求,为嵌入到移动设备中提供了有利条件。
  • 关键词

    矿用电铲目标检测轻量化YOLOv7GhostNetGSConv坐标注意力机制双向特征金字塔

  • 文章目录
    0 引言
    1 YOLOv7模型
    2 改进YOLOv7模型
    2.1 GhostNet网络
    2.2 GSConv模块
    2.3 CA模块
    2.4 BiFPN
    3 实验结果与分析
    3.1 实验平台
    3.2 数据集建立
    3.3 不同轻量化模块不同位置对比实验
    3.4 不同注意力机制模块对比实验
    3.5 模型对比实验
    3.6 实际场景检测效果对比实验
    4 结论
相关问题

主办单位:煤炭科学研究总院有限公司 中国煤炭学会学术期刊工作委员会

©版权所有2015 煤炭科学研究总院有限公司 地址:北京市朝阳区和平里青年沟东路煤炭大厦 邮编:100013
京ICP备05086979号-16  技术支持:云智互联