基于特征重组和注意力机制的YOLOv5光伏面板缺陷检测方法  

YOLOv5 Photovoltaic Panel Defect Detection Method Based on Feature Recombination and Attention Mechanism

作　　者：武果[1] 高守正 张德福 肖国龙[1] 龚正平[1] 覃健 WU Guo;GAO Shouzheng;ZHANG Defu;XIAO Guolong;GONG Zhengping;QIN Jian(Guangzhou Electronic Technology Co.,Ltd.,CAS,Guangzhou 510070,China;School of Automation Science and Engineering,South China University of Technology,Guangzhou 510641,China;Guangzhou Institute of Technology,College of Intelligent Manufacturing and Electrical Engineering,Guangzhou 510800,China)

机构地区：[1]中科院广州电子技术有限公司,广东广州510070 [2]华南理工大学自动化科学与工程学院,广东广州510641 [3]广州理工学院智能制造与电气工程学院,广东广州510800

出　　处：《自动化与信息工程》2024年第5期47-53,共7页Automation & Information Engineering

摘　　要：针对光伏面板缺陷检测存在的小目标检测精度低、泛化能力差等问题,提出基于特征重组和注意力机制的YOLOv5光伏面板缺陷检测方法。首先,分别使用卷积注意力模块和压缩激励模块,提取目标图像在通道和空间维度上的特征,使网络更有效地学习目标特征;然后,使用感受野扩展模块进行特征融合,确保网络能够结合大感受野特征和微小特征进行综合判断;最后,引入特征重组模块,对不同尺寸的特征图进行特征重组,解决了网络对小目标信息缺失和不敏感的问题。通过消融实验和对比实验,验证了该方法的有效性。Aiming at the problems of low small object detection accuracy and poor generalization ability in defect detection of photovoltaic panels,a YOLOv5 photovoltaic panel defect detection method based on feature recombination and attention mechanism is proposed.Firstly,the convolutional attention module and compression excitation module are used separately to extract the features of the target image in the channel and spatial dimensions,enabling the network to learn the target features more effectively;Then,the receptive field extension module is used for feature fusion to ensure that the network can combine large receptive field features and small features for comprehensive judgment;Finally,a feature recombination module was introduced to perform feature recombination on feature maps of different sizes,solving the problem of missing and insensitive information for small targets in the network.The effectiveness of this method was verified through ablation experiments and comparative experiments.

关 键 词：特征重组 注意力机制 光伏面板 缺陷检测 YOLOv5 神经网络 

分 类 号：TP183[自动化与计算机技术—控制理论与控制工程]