Computer Engineering and Applications ›› 2024, Vol. 60 ›› Issue (3): 340-347.DOI: 10.3778/j.issn.1002-8331.2304-0390

• Engineering and Applications • Previous Articles     Next Articles

Defect Detection of Photovoltaic Modules Based on Multi-Scale Feature Fusion

TIAN Hao, ZHOU Qiang, HE Chenlong   

  1. School of Electrical and Control Engineering, Shaanxi University of Science and Technology, Xi’an 710000, China
  • Online:2024-02-01 Published:2024-02-01

基于多尺度特征融合的光伏组件缺陷检测

田浩,周强,贺晨龙   

  1. 陕西科技大学 电气与控制工程学院,西安 710000

Abstract: A photovoltaic modules defect detection algorithm based on multi-scale feature fusion is proposed to address the challenges of complex defect backgrounds, large differences in defect scales, and a high number of small target defects that traditional object detection algorithms cannot solve. The algorithm is based on the YOLOv5s framework. Firstly, a coordinate attention mechanism is embedded in the backbone network to extract important defect shapes and enhance the network’s feature extraction ability. Secondly, a bidirectional feature pyramid network is used in the Neck network to adaptively fuse image features of different scales using adaptive weights. Finally, a tiny target detection layer is added to the prediction layer, and the ASFF detection head is used to adaptively fuse different output layers to reduce the loss of target feature information. The improved algorithm is validated on a photovoltaic component dataset, and the experimental results show that it can quickly and accurately identify defects, with an mAP of 91.9% and a recall rate of 90.8%, which represents a 3.2 and 4.5 percentage points improvement in mAP and recall rate, respectively, compared to the YOLOv5s network.

Key words: photovoltaic modules, YOLOv5, defect detection, feature fusion

摘要: 针对光伏组件缺陷背景复杂,缺陷尺度差异较大,小目标缺陷较多等传统目标检测算法无法解决的问题,提出了一种基于多尺度特征融合的光伏组件缺陷检测算法。算法以YOLOv5s为框架,在主干网络中嵌入坐标注意力机制,用于提取重要的缺陷形态,增强网络特征提取能力;在颈部网络中使用双向特征金字塔,以自适应权重的方式融合不同尺度的图像特征;在预测层添加微小目标检测层,并结合ASFF检测头自适应融合不同输出层,减少目标特征信息丢失。在光伏组件数据集上进行验证,实验表明改进后的算法可以快速精准识别缺陷,其中mAP达到了91.9%,召回率达到了90.8%,相比于YOLOv5s网络,mAP和召回率分别提升了3.2、4.5个百分点。

关键词: 光伏组件, YOLOv5, 缺陷检测, 特征融合