EP-YOLO: A Printed Circuit Board Defect Detection Network Integrating Coordinate Attention and Multi-Level Gradient...

The dependence of electronic products on printed circuit boards (PCBs) is increasing. In recent years, PCB defect detection technology has achieved considerable results, but the defect targets are small, the background is complex, and there are many integrated components, which poses great challenges to product quality control. Therefore, this paper proposes a printed circuit board defect detection network that integrates coordinate attention and multi-level gradient flow optimization, called EP-YOLO, to improve the detection accuracy of printed circuit board defects. Firstly, this paper improves the model's ability to capture small target details by reconstructing the detection head of the small target. Based on this, a Shallow Context Feature Extraction (SCFE) network is designed to fuse shallow features with multi-scale features, effectively preventing the loss of shallow detail information and texture features. At the same time, in order to suppress background noise, this paper designs a multi-level feature gradient flow optimization module (C2f_CA, abbreviated as CCA) that integrates coordinate attention and a Cross-Stage Partial Frequency-domain Omni-Kernel module (CSPFOK) to enhance feature extraction capability. Finally, the SCYLLA-IoU (SIoU) optimization model training process was introduced. The experimental results showed that EP-YOLO achieved 97.5% mAP50 on the PKU-Market-PCB dataset and 98.5% mAP50 on the DeepPCB dataset, with a parameter reduction of approximately 12.55%, outperforming popular object detection networks. The results highlight the potential capabilities of EP-YOLO, providing a powerful and effective solution for industrial PCB defect detection.