23 February 2024 WDC-YOLO: an improved YOLO model for small objects oriented printed circuit board defect detection
Xudong Song, Yucheng Wang, Changxian Li, Lifang Song
Author Affiliations +
Abstract

The detection of surface defects on a printed circuit board (PCB) plays a crucial role in ensuring the quality of PCB products. To address the diversity of defect poses and the challenges associated with detecting small objects on PCB, this paper proposes an improved YOLOv7 model for small objects-oriented PCB defect detection. First, this paper improves the regression loss function of YOLOv7 by incorporating wise-IoU (WIoU) and replacing IoU with an outlier degree to develop a gradient-boosting allocation strategy, thereby increasing the network’s accuracy. Second, this paper proposes a coordinate attention dynamic mechanism that performs convolution operations with deformable convolutional networks v2 (DCNv2) using coordinate attention. This mechanism effectively suppresses redundant information. Finally, this paper proposes a dynamic head diverse (DyHead-d) module that prioritizes spatial awareness over scale awareness, building on DyHead. This module improves the network’s ability to localize small targets. Experimental results show that the WDC-YOLO achieves a mean average precision of 98.4% on public datasets, demonstrating a 3.1% improvement compared to the original network. The significantly enhanced detection accuracy meets the real-time requirements of PCB defect diagnosis, which is of great importance for quality control and cost reduction in PCB industrial production.

© 2024 SPIE and IS&T
Xudong Song, Yucheng Wang, Changxian Li, and Lifang Song "WDC-YOLO: an improved YOLO model for small objects oriented printed circuit board defect detection," Journal of Electronic Imaging 33(1), 013051 (23 February 2024). https://doi.org/10.1117/1.JEI.33.1.013051
Received: 13 November 2023; Accepted: 26 January 2024; Published: 23 February 2024
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Object detection

Defect detection

Small targets

Convolution

Feature extraction

Data modeling

Computer aided design

Back to Top