基于改进YOLOv8n的电力设备红外图像检测算法

Improved YOLOv8n-based Algorithm for Infrared Image Detection of Power Equipment

  • 摘要: 针对复杂环境下电力设备红外图像目标检测精度不高以及模型计算复杂度大的问题,提出一种基于改进YOLOv8n的电力设备红外图像检测算法。首先,使用可变形卷积瓶颈结构替换主干网络C2F模块中的瓶颈结构,加强模型对不同形状设备的特征提取能力。其次,引入坐标注意力,提升模型对目标的关注能力,降低背景干扰。然后,提出轻量级扩张式特征金字塔LDFPN替换原有颈部网络,保持模型识别精度的同时显著降低参数量和计算量。最后,引入SIoU损失函数,通过优化预测框与真实框的长宽比惩罚项,提升预测框的定位精度。实验结果表明,本文算法在电力设备红外数据集上平均精度均值为94.92%,相比原模型提升了2.87个百分点,参数量和计算量分别降低43.3%和24.7%,与其他主流检测算法相比也具有显著优势,对提升电力设备检测的自动化水平具有一定意义。

     

    Abstract: An algorithm based on an improved YOLOv8n model was investigated to address the challenges posed by the low detection accuracy and high computational complexity of power equipment detection using infrared images in complex environments. First, the bottleneck structures in the backbone network C2F module were replaced with deformable convolutional network version 3 bottleneck modules to enhance the ability of the model to extract the features of power equipment with various shapes. Second, the coordinate attention (CA) mechanism was integrated to enhance the focus of the model on the infrared images of the power equipment and reduce background interference. Subsequently, a lightweight dilated feature pyramid network was introduced to replace the original neck network, which significantly reduced the parameters and computational complexity of the model while maintaining the recognition accuracy. Finally, the SCYLLA intersection over union (SIoU) loss function was introduced to improve the localization accuracy of the predicted bounding boxes by optimizing the aspect ratio penalty between the predicted and ground truth boxes. In an experiment, the proposed algorithm achieved a mean average precision of 94.92% on a power-equipment infrared image dataset, which was a 2.87% improvement over the baseline model. Moreover, the number of parameters and computational complexity were reduced by 43.3% and 24.7%, respectively. Compared with other mainstream detection algorithms, the proposed method exhibited significant advantages, thereby contributing meaningfully to the advancement of automated power-equipment detection.

     

/

返回文章
返回