A small defect detection technique for industrial product surfaces based on the EA-YOLO model
摘要
The detection of surface defects in industrial products is vital for ensuring product quality. Ensuring real-time performance, while improving the detection accuracy of low-pixel-resolution small defects against background interference poses a significant challenge. To address this, the EA-YOLO model, based on Yolov8, is proposed. It includes three main improvements: replacing C2f (Faster Implementation of CSP Bottleneck with 2 convolutions) with a specially designed C2FN (Faster Implementation of CSP FastNet Block with 2 convolutions Network) in the backbone module to reduce parameters and GFLOPs, while enhancing speed; introducing the Environmental Awareness Dynamic Network (EADN) to prevent the loss of defect information in extreme positions; and using the improved Dynamic Adaptive Fusion Detector (DAF-Detect) for predictions. Case studies with the NEU-DET and PCB-DET datasets show that EA-YOLO achieves a mAP of 81.1 and 97.8%, respectively, improving by 4.3 and 4.1% compared to the baseline model, with reduced parameters, GFLOPs, and increased FPS, demonstrating good robustness and generalization ability.