Detection of coal gangue by YOLO deep learning method based on channel pruning

Deyong ShangZhibin LvZehua GaoYuntao Lia School of Mechanical Electrical & Information Engineering,China University of Mining and Technology,Beijing,Chinab Intelligent Mining and Robot Research Institute,China University of Mining and Technology,Beijing,China
DOI: https://doi.org/10.1080/19392699.2024.2331545
2024-03-25
International Journal of Coal Preparation and Utilization
Abstract:Rapid and accurate identification of coal and gangue is of great significance to improve sorting efficiency and reduce environmental pollution. Image recognition based on deep learning is the current research hotspot, but most of the existing deep learning network models for coal gangue recognition are complex in structure and rely on high-performance computing resources, which is difficult to deploy in edge devices. To resolve this problem, based on the YOLOv5s model and on the condition of ensuring the accuracy of coal gangue identification, this paper uses channel pruning and transfer learning technology to realize the lightweight of the coal gangue identification and detection model. Results show that the model size is significantly reduced by this method. The parameter and the amount of computation were reduced by 78.7% and 62%, and the reasoning speed was 6.8 ms per image. The superiority of this model is further verified by comparing it with 7 mainstream target detection networks. The proposed lightweight treatment method of coal gangue identification model effectively reduces the model volume and improves the real-time identification.
energy & fuels,mining & mineral processing
What problem does this paper attempt to address?