RRBM-YOLO: Research on Efficient and Lightweight Convolutional Neural Networks for Underground Coal Gangue Identification

Abstract

Coal gangue identification is the primary step in coal flow initial screening, which mainly faces problems such as low identification efficiency, complex algorithms, and high hardware requirements. In response to the above, this article proposes a new “hardware friendly” coal gangue image recognition algorithm, RRBM-YOLO, which is combined with dark light enhancement. Specifically, coal gangue image samples were customized in two scenarios: normal lighting and simulated underground lighting with poor lighting conditions. The images were preprocessed using the dim light enhancement algorithm Retinexformer, with YOLOv8 as the backbone network. The lightweight module RepGhost, the repeated weighted bi-directional feature extraction module BiFPN, and the multi-dimensional attention mechanism MCA were integrated, and different datasets were replaced to enhance the adaptability of the model and improve its generalization ability. The findings from the experiment indicate that the precision of the proposed model is as high as 0.988, the [email protected](%) value and [email protected]:0.95(%) values increased by 10.49% and 36.62% compared to the original YOLOv8 model, and the inference speed reached 8.1GFLOPS. This indicates that RRBM-YOLO can attain an optimal equilibrium between detection precision and inference velocity, with excellent accuracy, robustness, and industrial application potential.

Keywords

• coal gangue recognition
• YOLOv8
• lightweight module RepGhost
• attention mechanism MCA
• low light enhancement