关键词:
煤矸石识别
YOLOv8
深度学习
注意力机制
轻量化网络
摘要:
在煤炭开采过程中,对煤矸进行有效识别不仅可以提高煤炭的燃烧质量,还可以降低污染物排放,现有传统的带式输送机煤矸识别技术操作复杂、效率低、成本高。针对上述问题,提出一种基于YOLOv8-ER的带式输送机煤矸目标检测模型。该模型在YOLOv8基础模型上,引入高效通道注意力网络,对YOLOv8的主干网络进行优化,以增强图像中的重要信息、抑制次要信息,加强主干网络的特征提取能力,降低模型计算复杂度,提升模型学习效率和检测精度。该模型还引入结构重参化技术,以简化推理过程中的网络架构,进一步降低网络体积,增强模型在深层网络中的训练稳定性,提升模型推理速度。实验结果表明,YOLOv8-ER在处理速度和检测精度上均有显著提升,实现了240.8帧/s的高速检测,m AP@0.5达到了92.3%的高准确率;与传统YOLOv8模型相比,准确率提高了3.2%,m AP@0.5提高了4.4%,PR平衡率提高了2.6%,浮点运算次数和GPU使用量分别降低了23.2%和18.6%。根据煤矸可视化结果可以得出,YOLOv8-ER模型能更精准地识别煤矸,避免误检漏检现象,且置信度更高。