Sui Yiping, Zhang Lei, Sun Zhipeng, Yi Weixun, Wang Meng
College of Coal Engineering, Shanxi Datong University, Datong 037003, China.
Key Laboratory of Deep Coal Mining of the Ministry of Education, School of Mines, China University of Mining and Technology, Xuzhou 221116, China.
Sensors (Basel). 2024 Jan 11;24(2):456. doi: 10.3390/s24020456.
The recognition technology of coal and gangue is one of the key technologies of intelligent mine construction. Aiming at the problems of the low accuracy of coal and gangue recognition models and the difficult recognition of small-target coal and gangue caused by low-illumination and high-dust environments in the coal mine working face, a coal and gangue recognition model based on the improved YOLOv7-tiny target detection algorithm is proposed. This paper proposes three model improvement methods. The coordinate attention mechanism is introduced to improve the feature expression ability of the model. The contextual transformer module is added after the spatial pyramid pooling structure to improve the feature extraction ability of the model. Based on the idea of the weighted bidirectional feature pyramid, the four branch modules in the high-efficiency layer aggregation network are weighted and cascaded to improve the recognition ability of the model for useful features. The experimental results show that the average precision mean of the improved YOLOv7-tiny model is 97.54%, and the FPS is 24.73 f·s. Compared with the Faster-RCNN, YOLOv3, YOLOv4, YOLOv4-VGG, YOLOv5s, YOLOv7, and YOLOv7-tiny models, the improved YOLOv7-tiny model has the highest recognition rate and the fastest recognition speed. Finally, the improved YOLOv7-tiny model is verified by field tests in coal mines, which provides an effective technical means for the accurate identification of coal and gangue.
煤矸识别技术是智能矿山建设的关键技术之一。针对煤矿工作面低光照、高粉尘环境导致煤矸识别模型准确率低、小目标煤矸识别困难等问题,提出了一种基于改进YOLOv7-tiny目标检测算法的煤矸识别模型。本文提出了三种模型改进方法。引入坐标注意力机制以提高模型的特征表达能力;在空间金字塔池化结构后添加上下文变换器模块以提高模型的特征提取能力;基于加权双向特征金字塔的思想,对高效层聚合网络中的四个分支模块进行加权级联,以提高模型对有用特征的识别能力。实验结果表明,改进后的YOLOv7-tiny模型的平均精度均值为97.54%,帧率为24.73 f·s。与Faster-RCNN、YOLOv3、YOLOv4、YOLOv4-VGG、YOLOv5s、YOLOv7和YOLOv7-tiny模型相比,改进后的YOLOv7-tiny模型具有最高的识别率和最快的识别速度。最后,通过煤矿现场测试对改进后的YOLOv7-tiny模型进行了验证,为煤矸的准确识别提供了有效的技术手段。