改进YOLOv8n的光伏电池缺陷检测算法

Improved YOLOv8n Photovoltaic Cell Defect Detection Algorithm

  • 摘要: 针对光伏电池缺陷检测中,由于缺陷尺度差异、形态复杂及缺陷种类多样化等因素导致目标定位不准、漏检和误检的问题,提出一种改进YOLOv8n的光伏电池缺陷检测算法。首先,构建C2F_DCNv4模块作为主干和颈部网络的部分特征提取模块,利用其动态提取缺陷特性及内存访问优化能力,增强网络特征提取能力和处理速度。其次,设计ESPPM模块替换原空间金字塔池化模块,通过重新校准和激活通道特征与全局特征,减少局部信息丢失,增强网络对关键特征的捕捉能力。最后,设计CPEN(Central Penalized EIoU with NWD Loss)损失函数作为边界框损失函数,强化目标定位,进一步优化网络的收敛性和检测精度。实验结果表明:将改进的YOLOv8n在数据集上进行实验,相较原算法mAP@0.5提高2.1%;计算量降低0.3GFLOPs;参数量降低0.15M;检测速度达到128FPS,综合性能满足光伏电池缺陷检测要求。

     

    Abstract: In order to solve the problems of inaccurate target positioning, missed detection and false detection due to factors such as difference in defect scale, complex shape and diversified defect types in photovoltaic cell defect detection, an improved YOLOv8n photovoltaic cell defect detection algorithm was proposed. Firstly, the C2F_DCNv4 module is constructed as part of the feature extraction module of the backbone and neck network, and its dynamic defect extraction characteristics and memory access optimization capabilities are used to enhance the feature extraction ability and processing speed of the network. Secondly, the ESPPM module is designed to replace the original space pyramid pooling module, which reduces the local information loss and enhances the network's ability to capture key features by recalibrating and activating channel features and global features. Finally, the CPEN (Central Penalized EIoU with NWD Loss) loss function was designed as the bounding box loss function to strengthen the target positioning and further optimize the convergence and detection accuracy of the network. The experimental results show that compared with the original mAP@0 algorithm, the improved YOLOv8n is improved by 2.1%, the computational cost is reduced by 0.3GFLOPs, the parameter quantity is reduced by 0.05M, and the detection speed reaches 128FPS, which meets the requirements of photovoltaic cell defect detection.

     

/

返回文章
返回