Detection of coal and gangue based on improved YOLOv5.1 which embedded scSE module*

Accurate identification of coal-gangue have great significance for separation of coal-gangue. The traditional coal gangue identification method has the disadvantages of low accuracy and slow speed. Therefore, an intelligent classification method of coal-gangue based on multispectral imaging technology and target detection is proposed in this paper. According to the model structure of YOLOv5, add scSE module in CSPDarknet and CSP module. The improved YOLOv5 is referred to as YOLOv5.1. To begin with, the multispectral data of coal-gangue are collected, and the collected coal-gangue images are screened. Beside, three bands with high recognition rate and correlation are selected from 25 bands to form pseudo-RGB images. Otherwise, the RGB image of coal-gangue was detected by theYOLOv5.1. By detecting the separated single band, the recognition rate and correlation of band 6, 10 and 12 are higher. The experimental results show that the average accuracy of detecting coal-gangue in the test set reaches 98.34 %, and the detection time is about 3.62 s by using the model of YOLOv5.1 to train the RGB image of coal-gangue. This method can not only accurately identify coal-gangue, but also obtain the relative position of coal-gangue, which can be effectively used for coal-gangue identification.

Automated summary

This study proposes an intelligent coal-gangue classification method based on multispectral imaging technology and target detection, which achieves high accuracy and speed in identifying coal-gangue. By collecting multispectral data, selecting bands with high recognition rate, and training with the improved YOLOv5 model, accurate identification and relative positioning of coal-gangue are achieved.