YOLO-WAD for Small-Defect Detection Boost in Photovoltaic Modules

The performance of photovoltaic modules determines the lifetime of solar cells; however, accurate detection remains a challenge when facing smaller defects. To address this problem, in this paper, we propose a YOLO-WAD model based on YOLOv10n. Firstly, we replace C2f (CSP bottleneck with two convolutions) with C2f-WTConv (CSP bottleneck with two convolutions-wavelet transform convolution) in the backbone network to enlarge the receptive field and better extract the features of small-target defects (hot spots). Secondly, an ASF structure is introduced in the neck, which effectively fuses the different levels of output features extracted by the backbone network and enhances the model's ability to detect small objects. Subsequently, an additional detection layer is added to the neck, and C2f is replaced by C2f-EMA (CSP bottleneck with two convolutions-efficient multi-scale attention mechanism), which can redistribute feature weights and prioritize relevant features and spatial details across image channels to improve feature extraction. Finally, the DyHead (dynamic head) detection head is introduced, which enables comprehensive scale, spatial, and channel awareness. This greatly enhances the model's ability to classify and localize small-target defects. The experimental results show that YOLO-WAD detects our dataset with an overall accuracy of 95.6%, with the small-target defect detection accuracy reaching 86.3%, which is 4.1% and 9.5% higher than YOLOv10n and current mainstream models, verifying the feasibility of our algorithm.