Research on lightweight Yolo coal gangue detection algorithm based on resnet18 backbone feature network

Abstract

Nowadays coal gangue presorting is still mostly conducted manually, with high labor intensity, low sorting efficiency, and potential safety hazards. A coal gangue sorting robot that is an effective device to complete the coal gangue presorting instead of the worker, for which the coal gangue intelligent detection is one of its key technologies. The lightweight and real-time performance of the detection model has an important impact on the performance of coal gangue sorting robot. Lots of models for the coal gangue detection have been proposed based on machine learning, but there are still some problems such as slow detection speed, structural redundancy and large model size. This paper adopts the lightweight network of ResNet18 as the backbone feature extraction network of YOLOv3 and proposes a coal gangue detection algorithm of ResNet18-YOLO, studied the feature scale reduction and unstructured pruning of the model to further improve its lightweight and real-time performance, prepared the coal gangue data set referring to the actual gangue sorting situation and discussed the performance of the model. The results show that the improved ResNet18-YOLO algorithm can detect the coal gangue at a speed of 45.5 ms/piece and the model size is only approximately 65.34 MB when the mAP of the coal gangue detection is 96.27%. It has better real-time performance and smaller model size with the condition of equivalent mAP performance to the other algorithms, and has good gangue detection performance, which is conducive to reducing the technical requirements for the gangue sorting robot.