HLG-YOLOv7: Small object detection in conveyor belt damage based on leveraging hybrid local and global features

Abstract

In the industrial production process, the detection of conveyor belt damage plays a crucial role in ensuring the stable operation of the transportation system. To tackle the issues of significant changes in damage size, missed detections, and poor detection ability of small-size objects in conveyor belt surface damage detection, an improved HLG-YOLOv7 (Hybrid Local and Global Features Network) conveyor belt surface defect detection algorithm is proposed. Firstly, Next-VIT is employed as the backbone network to extract local and global features of the damage, enhancing the model's ability to extract features of different-sized damages. Additionally, to deeply utilize the extracted local and global features, the Explicit Visual Center (EVC) feature fusion module is introduced to obtain comprehensive and discriminative feature representations, further enhancing the detection capability of small objects. Lastly, a lightweight neck structure is designed using GSConv to reduce the complexity of the model. Experimental results demonstrate that the proposed method performs better at detecting small objects than existing methods. The improved algorithm achieves mAP and F1 scores of 96.24% and 97.15%, respectively, with an FPS of 28.2.