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

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 mAP@0.5(%) value and mAP@0.5: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.