基于改进YOLO v5s算法的光伏组件故障检测

Photovoltaic Module Fault Detection Based on Improved YOLOv5s Algorithm

  • 摘要: 针对无人机在光伏组件巡检任务中红外故障图像识别准确率低、检测速度慢的问题,提出一种特征增强的YOLO v5s故障检测算法。首先对损失函数进行优化,将原有的回归损失计算方法由GIOU(generalized intersection over union)改为功能更加强大的EIOU(efficient intersection over union)损失函数,并自适应调节置信度损失平衡系数,提升模型训练效果;随后,在每个检测层前分别添加InRe特征增强模块,通过丰富特征表达增强目标特征提取能力。最后,用创建的红外光伏数据集进行对比验证。实验结果表明:本文方法均值平均精度(mean average precision, mAP)为92.76%,检测速度(frame per second,FPS)达到42.37 FPS,其中热斑、组件脱落两种故障类型平均精度分别为94.85%、90.67%,完全能够满足无人机自动巡检的需求。

     

    Abstract: Infrared fault images have the limitations of a low recognition accuracy and low detection rate in a PV module inspection task using a UAV. To address these issues, a feature enhanced YOLO v5s fault detection algorithm is proposed. First, the loss function is optimized, the original regression loss calculation method is changed from GIOU to EIOU, and the confidence loss balance coefficient is adjusted adaptively to improve the model training. The InRe feature enhancement module is then added before each detection layer to enhance the ability of the target feature extraction by enriching the feature expression. Finally, comparative experiments are conducted using the infrared photovoltaic dataset created in this study. The experimental results show that the detection mAP of our method is 92.76%, whereas the detection speed is 42.37 FPS. The mean average precisions of the hot spot and component falling off were 94.85% and 90.67%, respectively, which can fully meet the requirements of the automatic inspection of the UAV.

     

/

返回文章
返回