基于改进YOLOv5的煤矸识别研究

作者：张释如; 黄综浏^*; 张袁浩; 章鳌; 季亮

来源：工矿自动化, 2022, 48(11): 39-44.

DOI：10.13272/j.issn.1671-251x.2022060052

• 煤矸识别
• 小目标煤矸
• YOLOv5
• 深度学习
• 空洞卷积
• 残差ASPP
• AdaBelief优化算法

现有基于深度学习的煤矸识别方法应用于井下复杂环境中时易出现误检和漏检情况，且对小目标煤矸的识别精度低。针对该问题，提出一种改进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模型。

• 单位
  西安科技大学