基于改进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: Aiming at the commonly used computer vision-based PV panel defect recognition model to extract features difficult, 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 improved model adaptability by dynamically adjusting attention span; designing a feature extraction network incorporating the SE attention mechanism, which improves the sensitivity of the network to key features, thus improving the feature characterization; Better Student semi-supervised learning method(BSLM) is proposed, which improves the model accuracy while reducing the amount of data labeling. Experimentally, it is proved that the mAP@0.5 of the algorithm proposed in this paper reaches 83.9%, which is 3.1% higher than that of YOLOv8n algorithm, far exceeding other one-stage algorithms, and it can better meet the accuracy requirements of defect detection of PV panels. The FPS reaches 101.01, which meets the real-time requirements of defect detection of PV panels. After using the BSLM proposed in this paper, the mAP@0.5 reaches 90.7%, which further improves the model performance.

     

/

返回文章
返回