Abstract:Aiming at the problems of long detection time and low detection accuracy in the existing coal gangue recognition, this paper proposes a method to collect the multispectral images of coal gangue using spectral technology and match with the improved YOLOv5s (You Only Look Once Version-5s) neural network model to apply it to coal gangue target recognition and detection, which can effectively reduce the detection time and improve the detection accuracy and recognition effect of coal gangue. In order to take the coverage area, center point distance and aspect ratio into account at the same time, the improved YOLOv5s neural network replaces the original GIou Loss loss function with CIou Loss loss function. At the same time, DIou NMS replaces the original NMS, which can effectively detect overlapping targets and small targets. In the experiment, 490 sets of multispectral data were obtained through the multispectral data acquisition system. Using the random forest algorithm and the correlation analysis of bands, the spectral images of the sixth, twelfth and eighteenth bands from twenty-five bands were selected to form a pseudo RGB image. A total of 974 original sample images of coal and gangue were obtained. Through two image noise reduction methods, namely, Gaussian filtering algorithm and non-local average noise reduction, 1948 images of coal gangue were obtained after preprocessing the dataset. This was divided into a training set and test set according to an 8:2 ratio and trained in the original YOLOv5s neural network, improved YOLOv5s neural network and SSD neural network. By identifying and detecting the three neural network models obtained after training, the results can be obtained, the loss value of the improved YOLOv5s neural network model is smaller than the original YOLOv5s neural network and SSD neural network, the recall rate is closer to 1 than the original YOLOv5s neural network and SSD neural network, the detection time is the shortest, the recall rate is 100% and the average detection accuracy of coal and gangue is the highest. The average precision of the training set is increased to 0.995, which shows that the improved YOLOv5s neural network has a better effect on the detection and recognition of coal gangue. The detection accuracy of the improved YOLOv5s neural network model test set is increased from 0.73 to 0.98, and all overlapping targets can also be accurately detected without false detection or missed detection. At the same time, the size of the improved YOLOv5s neural network model after training is reduced by 0.8 MB, which is conducive to hardware transplantation.

A Data Expansion Strategy for Improving Coal-Gangue Detection

Enhancing Coal-Gangue Object Detection Using GAN-based Data Augmentation Strategy with Dual Attention Mechanism

A fast-training GAN for coal–gangue image augmentation based on a few samples

Optimization Study of Coal Gangue Detection in Intelligent Coal Selection Systems Based on the Improved Yolov8n Model

High Quality Coal Foreign Object Image Generation Method Based on StyleGAN-DSAD

A high accuracy detection method for coal and gangue with S3DD-YOLOv8

Research of coal and gangue identification and positioning method at mobile device

Improved YOLOv7 Network Model for Gangue Selection Robot for Gangue and Foreign Matter Detection in Coal

Coal gangue detection and recognition algorithm based on deformable convolution YOLOv3

STATNet: One-stage Coal-Gangue Detector Based on Deep Learning Algorithm for Real Industrial Application

Data-driven Model SSD-BSP for Multi-Target Coal-Gangue Detection

Target Recognition of Coal and Gangue Based on Improved YOLOv5s and Spectral Technology

Detecting Coal Content in Gangue Via Machine Vision and Genetic Algorithm-Backpropagation Neural Network

A fast recognition method for coal gangue image processing

Image Recognition of Coal and Coal Gangue Using a Convolutional Neural Network and Transfer Learning

Dual-Path GAN: A Method for Enhancing Small-scale Defect Detection on Metal Images

A fast gangue detection algorithm based on multi-head self-attention mechanism and anchor frame optimization strategy

Research on efficient classification algorithm for coal and gangue based on improved MobilenetV3-small

Multiple factors influence coal and gangue image recognition method and experimental research based on deep learning

Research on improved gangue target detection algorithm based on Yolov8s

Research on intelligent detection of coal gangue based on deep learning