基于改进YOLOv8的光伏组件缺陷红外图像检测算法

Infrared Image Defect Detection Algorithm for Photovoltaic Modules Based on Improved YOLOv8

  • 摘要: 无人机航拍的光伏组件红外图像存在着对比度弱、缺陷形状各异、尺度变化大等问题。针对上述问题提出了基于改进YOLOv8的光伏组件缺陷检测算法。首先,在Backbone和Neck中分别引入InceptionNext模块,通过分解大核深度卷积获取更加丰富的上下文信息;其次,引入SPD(space-to-depth)模块,通过空间到深度方向的特征重组实现下采样,减少特征信息的丢失,加强对小目标缺陷的特征学习能力;最后,在Neck中引入极化自注意力(Polarized self-attention)机制,捕捉不同尺度特征层上的关键特征信息,自适应调整不同通道组之间的权重,增强特征表示能力。实验结果表明,所提算法的mAP50达到79.2%,mAP50:95达到44.5%,较基线模型分别提高了8.6%、10%,能够有效提高光伏组件缺陷的检测精度。

     

    Abstract: There are some problems in the infrared images of photovoltaic modules taken by UAVs, such as weak contrast, different defect shapes, and large scale changes. In order to solve the above problems, a defect detection algorithm for photovoltaic modules based on improved YOLOv8 was proposed, firstly, the InceptionNext module was introduced into Backbone and Neck, respectively, to obtain richer context information by decomposing the deep convolution of large kernels, secondly, the space-to-depth (SPD) module was introduced to realize downsampling through feature recombination in the space-to-depth direction, so as to reduce the loss of feature information and strengthen the feature learning ability of small target defects, and finally, Polarized was introduced into Neck self-attention) mechanism, which captures the key feature information on the feature layers of different scales, adaptively adjusts the weights between different channel groups, and enhances the feature representation ability. Experimental results show that the mAP50 and mAP50:95 of the proposed algorithm reach 79.2% and 44.5%, which are 8.6% and 10% higher than those of the baseline model, respectively, indicating that the proposed algorithm can effectively improve the detection accuracy of infrared image defects of photovoltaic modules.

     

/

返回文章
返回