-
题名基于改进YOLOv5的煤矸识别研究
被引量:5
- 1
-
-
作者
张释如
黄综浏
张袁浩
章鳌
季亮
-
机构
西安科技大学通信工程学院
中煤科工集团常州研究院有限公司
天地(常州)自动化股份有限公司
-
出处
《工矿自动化》
北大核心
2022年第11期39-44,共6页
-
基金
国家自然科学基金资助项目(51774234)
陕西省榆林市科技计划项目(CXY-2020-035)
天地科技股份有限公司科技创新创业资金专项项目(2020-TD-ZD010)。
-
文摘
现有基于深度学习的煤矸识别方法应用于井下复杂环境中时易出现误检和漏检情况,且对小目标煤矸的识别精度低。针对该问题,提出一种改进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
深度学习
空洞卷积
残差ASPP
AdaBelief优化算法
-
Keywords
coal and gangue recognition
small coal and gangue target
YOLOv5
deep learning
atrous convolution
residual ASPP
AdaBelief optimization algorithm
-
分类号
TD67
[矿业工程—矿山机电]
-