基于改进YOLOv8及BSLM的光伏板红外图像检测

Infrared Image Detection of Photovoltaic Panels Based on Improved YOLOv8 and BSLM

  • 摘要: 针对常用的基于计算机视觉的光伏板缺陷识别模型提取特征困难、推理速度慢等问题,本文提出了一种以融合双层路由注意力机制视觉转换器Biformer为骨干网络的YOLOv8改进算法,通过动态调整注意力范围,提高了模型的适应能力;设计了融合SE注意力机制的特征提取网络,提高了网络对关键特征的敏感度,从而提高了特征的表征能力;提出了Better Student半监督学习方法(better student learning method, BSLM),在数据标注量不变的同时提高了模型精度。实验证明,改进YOLOv8算法的mAP@0.5达到了83.9%,比YOLOv8n算法提高了3.1%,远超其他一阶段算法,能更好地满足光伏板缺陷检测的精度要求。FPS达到了101.01,满足对光伏板缺陷检测的实时性要求。在使用本文提出的BSLM后,mAP@0.5达到了90.7%,进一步提升了模型性能。

     

    Abstract: Aimed at the commonly used computer vision-based photovoltaic (PV) panel defect recognition model to extract difficult features, slow inference speed, and other problems, this paper proposes a YOLOv8 improvement algorithm that incorporates a two-layer routing attention mechanism visual converter biformer as the backbone network, which improves the model adaptability by dynamically adjusting the attention span. A feature extraction network is designed incorporating the squeeze-and-excitation attention mechanism, which improves the sensitivity of the network to key features, thus improving the feature characterization. The better student semi-supervised learning method (BSLM) is proposed, which improves the model accuracy while reducing the amount of data labeling. Experimentally, it was proven that the mean average precision at the intersection over union threshold of 0.5 (mAP@0.5) of the proposed algorithm reached 83.9%, which is 3.1% higher than that of the YOLOv8n algorithm, far exceeding other one-stage algorithms. It can also better meet the accuracy requirements of defect detection in PV panels. The frames per second reached 101.01, which satisfies the real-time requirements of defect detection in PV panels. After using the BSLM proposed in this paper, the mAP@0.5 reached 90.7%, which further improved the model performance.

     

/

返回文章
返回