基于CA-YOLOv8的输变电设备红外图像检测算法

Infrared Image Detection Algorithm for Power Transmission and Transformation Equipment Based on CA-YOLOv8

  • 摘要: 在变电站的实际工作环境当中,输变电设备的红外图像会由于不同的拍摄场景和光照强度等问题影响最终的数据质量,对红外图像中输变电设备的定位及识别造成很大的影响。为了解决这个问题,提出一种基于CA-YOLOv8的目标检测算法,它通过在YOLOv8的网络结构中添加经过改进的CA模块以提高骨干网路的特征提取能力;同时,模型采用SIoU作为回归损失,既解决CIoU的梯度消失问题,又减小自由度的总数,提高模型的收敛速度。在实测数据集上采用Grad-CAM++对模型进行视觉可解释性展示和实验验证,本文算法比目前主流的YOLO系列算法具有更高的准确率,并通过损失训练过程和实际预测结果验证了本文算法能够快速准确地实现输变电设备的识别和定位。

     

    Abstract: Infrared imaging of power transmission and transformation equipment in substations is often influenced by varying shooting scenes and light intensities, which degrades data quality. This has a significant impact on the positioning and identification of power transmission and transformation equipment in infrared images. To address this problem, an object detection algorithm based on CA-YOLOv8 is proposed, which enhances feature extraction in the backbone network by integrating an improved Coordinate Attention module into the YOLOv8 architecture. Furthermore, by employing SIoU as the regression loss, the algorithm overcomes the gradient vanishing issue of IoU and reduces degrees of freedom, thereby accelerating model convergence. Grad-CAM++ was applied to visualize and verify the model performance on the measured dataset. The algorithm demonstrates higher accuracy compared to mainstream YOLO series algorithms. Based on the loss training process and actual prediction results, the algorithm enables rapid and accurate identification and localization of power transmission and transformation equipment.

     

/

返回文章
返回