基于改进YOLOv8n的轻量化煤矸识别算法研究

Research on Lightweight Coal Gangue Recognition Algorithm Based on Improved YOLOv8n

  • 摘要: 针对传统煤矸检测算法容易出现漏检、误检,且在计算资源受限的条件下难以实现高效部署等问题,提出了一种基于改进YOLOv8n的轻量化煤矸识别算法。在YOLOv8n算法的基础上,采用扩张残差结构DWR (Dilation-Wise Residual)改进主干网络部分C2f模块,提高目标识别的准确性;引入加权双向特征金字塔网络BiFPN,提升检测能力并降低参数量;设计一种轻量级检测头LSCSBD (Lightweight Shared Convolutional Separate Batch Normalization Detection Head),进一步降低计算复杂度;引入Focaler-MPDIoU函数,提高了模型的检测精度。试验结果表明:改进YOLOv8n模型的运算量、参数量和权重大小相较于原模型分别下降26.8%、47.9%和46.0%,平均精度均值提升1.4%,降低了煤矸石漏检和误检问题,在保证模型轻量化的同时提高了检测精度。

     

    Abstract: Aiming at the problems that missed and false detections are prone to occur in traditional coal gangue detection algorithm, as well as the difficulty in achieving efficient deployment under the condition of limited computing resources, and others, a type of lightweight coal gangue recognition algorithm based on improved YOLOv8n is proposed. On the basis of YOLOv8n algorithm, the expansion residual structure DWR (Dilation-Wise Residual) is adopted to improve the C2f module of the backbone network part, and the accuracy of target identification is improved; The weighted bidirectional feature pyramid network BiFPN is introduced to improve detection capability and reduce parameter quantity; A type of lightweight detection head LSCSBD(Lightweight Shared Convolutional Separate Batch Normalization Detection Head) is designed to further reduce the computational complexity; Introducing Focaler-MPDIoU function, the detection accuracy of the model is improved. The test results show that the computational load, parameter quantity, and weight size of the improved YOLOv8n model compared to the original model is reduced by 26.8%, 47.9%, and 46.0%, respectively. The average mean accuracy value is improved by 1.4%, which reduces the problems of missed and false detections of coal gangue and improves the detection accuracy while ensuring the model lightweighting.

     

/

返回文章
返回