Traditional infrared photovoltaic module defect detection methods often suffer from low accuracy and slow processing speeds. To address these limitations
we propose WD_YOLOv8
an improved infrared photovoltaic module defect detection method based on YOLOv8. First
we introduced the Wise Intersection over Union v2 loss function to replace the Complete-IoU loss function used in the original algorithm
the v2 loss function improved the balance between precision () and recall ()
optimizing overall detection performance. Second
DySample up-sampling was incorporated into the feature fusion network (Neck) to maintain a high detection frame rate. Experimental results show that the improved model increases metrics
mAP@0.5
and F1-score by 0.3
1.5
and 2.6 percentage points
respectively
in the infrared pseudo-color photovoltaic module defect detection scenarios. Moreover
in infrared grayscale photovoltaic module defect detection scenarios
these metrics improve by 18.6
7.7
and 5.1 percentage points
respectively. These findings indicate that the proposed algorithm exhibits strong robustness and adaptability in infrared imaging applications.
关键词
Keywords
references
LIU W, ANGUELOV D, ERHAN D, et al.SSD: single shot MultiBox detector[C]//Computer Vision-ECCV 2016. Cham: Springer, 2016: 21-37.
REDMON J, DIVVALA S, GIRSHICK R, et al.You only look once: unified, real-time object detection[C]//2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR). Las Vegas, NV, USA, 2016: 779-788.
HE K M, GKIOXARI G, DOLLÁR P, et al. Mask R-CNN[C]//2017 IEEE International Conference on Computer Vision (ICCV). Venice, Italy, 2017: 2980-2988.
REN S Q, HE K M, GIRSHICK R, et al.Faster R-CNN: towards real-time object detection with region proposal networks[J]. IEEE transactions on pattern analysis and machine intelligence, 2017, 39(6): 1137-1149.
LIU W Z, LU H, FU H T, et al.Learning to upsample by learning to sample[C]//2023 IEEE/CVF International Conference on Computer Vision (ICCV). Paris, France, 2024: 6004-6014.
WANG J Q, CHEN K, XU R, et al.CARAFE: content-aware ReAssembly of FEatures[C]//2019 IEEE/CVF International Conference on Computer Vision (ICCV). Seoul, Korea, 2019: 3007-3016.