基于改进YOLOv5的煤矸识别研究
张释如1
黄综浏2
张袁浩3
章鳌2
季亮3
1.西安科技大学 通信工程学院,陕西 西安 7106002.西安科技大学 通信工程学院,陕西 西安 710600;中煤科工集团常州研究院有限公司,江苏 常州 2130153.中煤科工集团常州研究院有限公司,江苏 常州 213015;天地(常州)自动化股份有限公司,江苏 常州 213015
摘要：现有基于深度学习的煤矸识别方法应用于井下复杂环境中时易出现误检和漏检情况,且对小目标煤矸的识别精度低.针对该问题,提出一种改进YOLOv5模型,并基于该模型实现煤矸识别.对采集的煤与矸石数据进行数据增强,以丰富数据集,提高数据利用率;在空间金字塔池化(SPP)模块中引入空洞卷积和残差块,得到残差ASPP模块,可在不损失图像信息的前提下,增大卷积输出感受野,强化模型对深层特征的提取;采用AdaBelief优化算法代替YOLOv5原有的Adam优化算法,提高模型的收敛速度与识别精度.实验结果表明:AdaBelief优化算法和残差ASPP模块可有效提高YOLOv5模型的精确率、召回率和平均精度均值(mAP);改进YOLOv5模型的mAP达到94.43％,比原始YOLOv5模型提高了2.27％,帧率降低了0.03帧/s,性能优于SSD,Faster R-CNN,YOLOv3,YOLOv4等主流目标检测模型;在极端黑暗的环境中,改进YOLOv5模型也能准确划定目标边界,识别效果优于其他改进YOLOv5模型.
关键词：煤矸识别小目标煤矸YOLOv5深度学习空洞卷积残差ASPPAdaBelief优化算法
分类号：TD67(矿山电工)
资助基金：国家自然科学基金(51774234)陕西省榆林市科技计划项目(CXY-2020-035)天地科技股份有限公司科技创新创业资金专项项目(2020-TD-ZD010)
论文发表日期：2022-11-10
在线出版日期：2025-08-15（本平台首次上网日期，不代表文献的发表时间）
页数：6( 39-44 )
英文信息