• 论文
主办单位:煤炭科学研究总院有限公司、中国煤炭学会学术期刊工作委员会
基于改进YOLOv5s模型的煤矸目标检测
  • 作者

    沈科季亮张袁浩邹盛

  • 单位

    中煤科工集团常州研究院有限公司天地(常州)自动化股份有限公司

  • 摘要

    针对现有基于深度学习的煤矸目标检测方法存在检测速度慢且检测精度较低等问题,提出了一种改进YOLOv5s模型,并将其应用于煤矸目标检测中。改进YOLOv5s模型在YOLOv5s模型Backbone区域嵌入自校正卷积SCConv作为特征提取网络,可更好地融合多尺度特征信息;由于煤块和矸石的尺寸相对整张图像过小,对YOLOv5s模型Neck区域进行适当精简,将适合检测较大尺寸对象的19×19特征图分支删除,从而降低模型复杂度并提高检测实时性;对通过K-means算法聚类得到的锚框进行线性缩放,提高模型检测精度。基于改进YOLOv5s模型的煤矸目标检测实验表明,相较于YOLOv5s模型,改进YOLOv5s模型能准确检测出相应的煤块和矸石,且改进YOLOv5s模型大小降低了1.57 MB,帧速率增加了2.1帧/s,平均精度均值提高了1.7%,表明改进的YOLOv5s模型检测精度和检测速度均有提升。

  • 关键词

    煤矸分选煤矸识别目标检测YOLOv5s

  • 基金项目(Foundation)
    天地科技股份有限公司科技创新创业资金专项项目(2020-TD-ZD010)
  • 引用格式
    沈科,季亮,张袁浩,等.基于改进YOLOv5s模型的煤矸目标检测[J].工矿自动化,2021,47(11):107-114.
  • 相关文章
  • 相关专题
相关问题

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

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