基于改进YOLOv8模型的树线接地故障识别

王洪江 ,  刘金圣 ,  赵宏 ,  赵婷婷 ,  代钦 ,  高英才

辽宁工程技术大学学报(自然科学版) ›› 2025, Vol. 44 ›› Issue (01) : 113 -119.

PDF (1116KB)
辽宁工程技术大学学报(自然科学版) ›› 2025, Vol. 44 ›› Issue (01) : 113 -119. DOI: 10.11956/j.issn.1008-0562.20240064
图像识别与处理

基于改进YOLOv8模型的树线接地故障识别

作者信息 +

Tree line grounding fault identification technology based on improved YOLOv8 model

Author information +
文章历史 +
PDF (1141K)

摘要

为提升电力系统中树线接地故障检测的识别效果,提出一种改进YOLOv8模型。该模型通过插入SimAM注意力机制增强特征表示能力,采用GIoU损失函数提升边界框预测的准确性,提高模型在复杂环境下的故障识别性能。为验证改进YOLOv8模型的性能进行消融实验、SimAM注意力机制模块的插入位置变化实验、损失函数选择实验,以及与其他识别模型的对比实验。实验结果表明:改进YOLOv8模型的识别精确度、召回率、平均精度均最高。该模型有效提高了树线接地故障检测图像的识别精度,为输电线路的智能化运维提供技术支持。

关键词

电力系统 / 树线接地故障 / YOLOv8模型 / SimAM注意力机制 / GIoU损失函数

Key words

power system / tree line grounding fault / YOLOv8 model / SimAM attention mechanism / GIoU loss function

引用本文

引用格式 ▾
王洪江,刘金圣,赵宏,赵婷婷,代钦,高英才. 基于改进YOLOv8模型的树线接地故障识别[J]. 辽宁工程技术大学学报(自然科学版), 2025, 44(01): 113-119 DOI:10.11956/j.issn.1008-0562.20240064

登录浏览全文

4963

注册一个新账户 忘记密码

0 引言

电力系统的稳定运行是现代社会的基石之一[1]。中国现行的6~10 kV配电网中,单相接地故障占系统故障的80%以上[2],其中,树线接地故障常见且难以预测。这种故障通常由树木与输电线路的接触引起,不仅会造成电力供应的中断,甚至还可能引发火灾等安全事故[3-5]。因此,实现对树线接地故障的快速、准确识别成为保障电网安全运行的关键。

近年来,深度学习模型在图像识别领域的突破性进展,为电力系统故障检测提供了新的技术路径。YOLOv8模型以其高效性和准确性被广泛应用于各类目标检测场景,该模型在通用目标检测任务上表现出色,但在复杂环境下的小目标检测方面仍存在局限性。例如,在电力系统故障识别中,YOLOv8模型对电力系统特有故障特征的学习不够充分,并且在复杂背景下对微小故障迹象的捕捉能力欠佳,导致识别精度受限。赖秋频等[6]将深度学习模型YOLOv2应用于无人机拍摄的输电线路绝缘子图像识别与诊断,通过训练YOLOv2模型,识别复杂背景下的绝缘子特征,结合多种图像处理方法进行缺陷诊断,有效提升了输电线路巡检的智能化水平和效率。郝帅等[7]针对复杂背景和遮挡问题,将卷积块注意模型融合于YOLOv3模型框架,增强故障目标的识别能力,并通过改进的非极大值抑制方法和Focal Loss函数减少漏检率,提高检测精度。郑伟等[8]针对输电线路维护中的典型缺陷识别问题,设计了一种基于改进YOLOv4的无人机实时检测模型,结合MobileNet和空洞模块优化特征提取网络,引入自适应路径融合网络增强多尺度目标检测,显著提升了检测精度和降低了误检率。郝帅等[9]针对复杂环境和多尺度特征与遮挡挑战,提出基于注意力机制与跨尺度特征融合的YOLOv5故障检测算法,通过空间与通道卷积注意力模型、BiFPN结构和自适应加权融合模块,提高了遮挡情况下算法的检测能力。

已有研究将YOLO系列模型应用于电力系统故障识别,但针对树线接地故障优化方面的研究较少,因此提出一种基于改进YOLOv8模型的电力视觉方法,提升复杂自然环境下树线接地故障的识别性能。首先,结合SimAM注意力机制,增强模型对关键特征的提取能力,提高对微小故障特征的敏感性。其次,引入GIoU损失函数,优化目标框的定位精度,减少边界框重叠不足导致的误差。最后,通过标注构建多样化的数据集进行消融实验与对比试验来验证改进模型的有效性。

1 模型改进

1.1 YOLOv8模型

YOLOv8模型采用轻量级网络结构和高效推理技术,识别准确性和运算速度都显著提升[10]。YOLOv8模型网络结构见图1图1中,输入图像先经过由多个卷积层组成的主干网络,提取图像的基础特征[11];再经过特征金字塔网络(FPN)、路径聚合网络(PAN)和颈部网络进行特征融合,增强对不同尺度目标的检测;最后,头部模块利用融合特征图进行目标分类和定位,生成精确的边界框和类别标签,确保检测结果的准确性[12]

1.2 SimAM模块

SimAM(simple parameter-free attention module)是一种受神经科学启发的无参数三维注意力机制,用于提升深度学习模型中特征提取的效果。网络的每个输出层都经过自适应批量归一化处理,以稳定训练过程并提高模型的泛化能力。SimAM以最小能量函数来识别更重要的神经元和从而获得更多信息特征[13],通过能量函数[14]表达神经元的重要程度。

能量函数为

e t w t , b t , y , w i = y t - t ^ 2 + 1 M - 1 i = 1 M - 1 y 0 - x ^ i 2

式中: t ^为输入特征图的同一通道中目标神经元t的线性变换的结果, t ^ = w t t + b t x ^ i为输入特征图中其他神经元xi 的线性变换的结果, x ^ i = w t x i + b tM为神经元个数[15] w t为目标神经元的线性变换权重; b t 为目标神经元的线性变换偏置;y为目标神经元t和其他神经元 x i的理想输出标签。

为简化计算,将目标神经元的标签 y t和其他神经元的标签 y 0设置为二元标签: y t = 1 y 0 = - 1

添加正则项λ,整合式(1)可得

e t w t , b t , y , w i = 1 M - 1 i = 1 M - 1 - 1 - w t x i + b t 2 +
1 - w t t + b t 2 + λ w t

式(2)分别对wtbt 求偏导,并令偏导为0,求解得wtbt

w t = - 2 t - u t t - u t 2 + 2 σ t 2 + 2 λ b t = - 1 2 t + u t w t

式中,utσt 分别为目标神经元以外神经元的均值与方差。

式(3)代入式(2)得最小能量

e t * = 4 ( σ t 2 + λ ) ( t - u t ) 2 + 2 σ t 2 + 2 λ

式(4)可知,最小能量 e t *越大说明目标神经元和周围神经元间的区别度越大,其值与重要性成正比[16]

根据注意力机制的定义,使用Sigmoid函数对特征进行增强处理,即

X ˜ = S i g m o i d 1 / E Х

式中: X ˜为处理后的特征;E 为所有空间和通道维度中最小能量的总和; X 为输入特征; 为逐元素相乘。

1.3 改进后的网络结构

根据YOLOv8原模型的网络架构,将SimAM注意力机制引入到网络第六层C2f模块之后,通过自适应批量归一化处理和能量函数调整特征图的注意力分布,提高模型对复杂空间信息和纹理细节的处理能力。改进后的YOLOv8模型网络结构见图2

1.4 GIoU损失函数

YOLOv8原始模型采用CIoU (complete intersection over union)损失函数[17-18],该函数在处理不同形状目标的相关性时,未考虑样本难易程度的区分问题。GIoU(generalized intersection over union)损失函数在保留IoU的尺度不变性的同时,进一步考虑了目标的非重叠区域,可以更准确描述不同形状目标之间的关系。此外,GIoU损失函数可以反映目标的难易程度,弥补CIoU损失函数在相关性描述和样本区分上的不足。因此,本文采用GIoU损失函数替换CIoU损失函数。

GIoU损失函数通过引入最小外接矩形的相关信息,改进了传统IoU在边界框不重叠时的不足。如图3所示,设C为预测框A与真实框B的最小外接矩形,先计算C的面积,然后计算外接矩形中不属于A、B框的区域面积占闭包区域面积的比[19],得出

I o U = A B A B G I o U = I o U - C - ( A B ) C

损失函数为

L o s s G l o U = 1 - G l o U

综上,GIoU通过引入在边界框不重叠时仍能提供梯度的度量,使边界框的位置可以优化,从而提高模型的性能和泛化能力[20]

2 实验结果及分析

2.1 数据集

针对树线接地故障识别优化问题,通过实地拍摄和搜集历史案例图片,选择具有多样性和真实性的528张高质量图像构建数据集。为提高模型学习的精确度,经过预处理及专业标注,精确标识每张图片的故障区域。将数据集按4∶1的比例分为训练集和验证集,训练集包含图片422张,验证集包含图片106张,确保模型训练的可靠性和泛化能力。数据集图片示例见图4

2.2 实验环境与设备

实验设备配置:Windows 11操作系统,CPU为Intel(R) Core(TM) i7-12650H,GPU为NVIDIA GeForce RTX 4060,深度学习框架使用PyTorch 2.1.2+CU121,代码编写与运行环境为PyCharm。

实验参数设置:训练轮次为300次,批量大小设置为16,图像输入尺寸(像素)为640×640,优化器采用SGD,并在训练过程中保持固定不变,学习率初始值为0.01。

2.3 评价指标

选择精确率P(precision)、召回率R(recall)、mAP_0.5、mAP_0.5:0.95作为评价指标对改进的YOLOv8模型进行性能评估。

P为模型预测的所有目标中正确预测目标的占比,可表示为

P = T P T P + F P

式中:TP为真正例,表示预测框与真实目标框IoU超过阈值的检测框数;FP为假正例,表示预测框与真实目标框IoU不超过阈值的检测框数。

R为模型正确预测的目标占真实目标的比,可表示为

R = T P T P + F N

式中,FN为错过的真实目标数[21]

mAP_0.5表示当IoU阈值设定为0.5时,所有类别平均精确度的平均值,反映改进YOLOv8模型的PR的变化趋势,该值越大表示模型越容易在高召回率下保持高精确率。

m A P _ 0.5 = 1 C k = 1 C A P _ 0.5

mAP_0.5:0.95是对模型性能的综合评估,为不同IoU阈值下的平均精确度(AP)的平均值。该指标能更全面地反映模型在不同准确性要求下的表现。本实验IoU阈值从0.5到0.95,步长为0.05。

2.4 消融试验及结果

为验证改进YOLOv8模型设计的轻量性与合理性,进行消融实验,对比不同模块的引入对模型性能的影响,结果见表1

表1可见,实验D在各项评价指标上均最优。相较于实验A,实验D的P提高了0.8个百分点,R提高了5.5个百分点,mAP_0.5、mAP_0.5:0.95分别提升了8.6个百分点和1.2个百分点。引入SimAM模块增强了模型的特征提取能力,使模型能够更有效地捕获目标的特征,特别是在复杂环境下对目标边界的精确定位得到了提升。单独使用SimAM模块,模型的精确率和召回率有所下降,这是因为引入的特征增强在特定层次对目标特征的分布产生了影响。实验C中,模型对边界框预测的准确性显著提高,R提升,但在未结合SimAM模块的情况下,对精确率的贡献有限。实验D中,SimAM模块与GIoU损失函数的结合发挥了协同作用,一方面通过SimAM增强了特征图对目标的区分能力,另一方面GIoU优化了边界框定位,因此在复杂背景下显著提高了目标检测的性能。实验结果表明,实验D的改进策略在目标检测任务中的表现优异,尤其适用于树线接地故障这类复杂而特殊的场景。

引入不同模块的改进YOLOv8模型在目标检测任务上的表现见图5。从图5中可见,随着训练轮次的增加,所有模型的mAP_0.5值均有所提高,其中,同时引入SimAM注意力机制模块和GIoU损失函数的改进YOLOv8模型明显优于其他模型,这说明该模型识别树线接地故障的准确性和有效性较高。

SimAM注意力机制模块在不同位置的插入对模型性能影响显著,在网络结构中选择合适的插入位置能够有效提升模型的目标检测能力。在YOLOV8n模型网络结构的不同层插入SimAM注意力模块,对比该模块的插入位置对改进模型性能的影响,结果见表2,其中+FLOOD7为本文提出的改进YOLOV8模型的网络结构。由表2图2可知,在第七层C2f模块后插入SimAM注意力机制模块(+FLOOR7),模型性能达到最佳,P为0.933,R为0.892,mAP_0.5和mAP_0.5:0.95分别为0.954和0.494。这是因为第七层是特征金字塔网络的关键位置,此位置既能够充分捕捉高层语义信息,又能够保留目标区域的细节特征,在此插入SimAM注意力机制模块能更有效地分配注意力,突出树线接地故障特征。在第五层C2f模块后插入SimAM注意力机制模块(+FLOOR5),模型性能有所下降。这是因为此位置较为靠近主干网络的浅层部分,特征图仍处于较低语义层次,导致SimAM注意力机制模块增强作用对高层语义信息的提取不足,限制了模型的整体检测性能。在第十层C2f模块后插入SimAM注意力机制模块(+FLOOR9),模型性能进一步下降。这是因为第十层靠近网络的后部,特征图已经过多次融合和下采样,语义信息趋于饱和,特征图细节损失较大,SimAM注意力机制模块难以显著提升注意力分布的有效性。

结合YOLOv8改进框架图可以看出,对于树线接地故障的识别SimAM模块的注意力的最佳插入位置在第七层C2f模块之后。这一位置能够充分利用主干网络提取的底层特征,同时为特征金字塔网络的高层特征提供增强支持,实现对树线接地故障特征的精准提取与识别。

2.5 对比试验与结果

(1)损失函数的对比实验

采用不同的损失函数,对改进YOLOV8模型进行实验,结果见表3。由表3可知,DIoU(distance intersection over union)损失函数通过最小化预测框和真实框中心点距离,加快模型的收敛速度,但未能有效处理长宽比的差异,因此其mAP_0.5:0.95值略小于GIoU损失函数。CIoU损失函数在DIoU损失函数的基础上进一步引入长宽比一致性作为惩罚项,提升了回归精度,但仍未能在非重叠情况下提供足够的优化信息,限制了其性能的进一步提升。EIoU(efficient intersection over union)损失函数和SIoU(scaled intersection over union)损失函数均细化了长宽比和方向惩罚项,但对于树线接地故障的特征复杂性较高的特定任务,没有使模型的mAP值提高。Shape IoU(shape intersection over union)损失函数强调预测框与真实框形状的相似性,但在目标边界复杂的场景下,其形状优化的作用相对有限。

综上,GIoU损失函数在目标检测性能方面显著优于其他损失函数。GIoU损失函数符合树线接地故障识别的场景,选择该函数作为损失函数显著提高了模型在定位和识别故障特征上的鲁棒性和准确性,是此类故障识别中优选损失函数。

(2)本文改进模型与其他模型的对比实验

为验证改进YOLOv8模型的准确性与合理性,分别采用改进YOLOv8模型、YOLOv8基本模型、YOLOv7-tiny模型、YOLOv5n系列模型、YOLOv3模型对树线接地故障场景进行识别,对比每种模型的mAP_0.5值和mAP_0.5:0.95值,结果见表4

表4可知,本文所提方法的识别准确率最高 。YOLOv5n基本模型的识别效果与YOLOv8n基本模型接近,但其引入SimAM注意力机制模块和GIoU损失函数后,其识别准确率的提升幅度不明显。这主要是因为YOLOv8n的架构设计更适合SimAM模块和GIoU损失函数的整合优化。改进YOLOv8n模型在树线接地故障识别任务中表现出色,优于其他主流模型。这主要是因为:① YOLOv8n通过引入更深层次的特征金字塔网络和改进的路径聚合网络,增强了对多尺度目标的检测能力;②YOLOv8n采用了新的归一化方法,提高了模型的非线性表达能力和训练稳定性。这些改进使得YOLOv8n在树线接地故障识别任务中,能够更准确地捕捉故障特征,降低漏检率和误检率。

3 结论

(1)在树线接地故障识别任务中,改进YOLOv8模型的识别精度、召回率、mAP_0.5值和mAP_0.5:0.95值均得到提升,验证了模型在故障检测任务中的有效性。通过消融实验,SimAM注意力机制模块的插入位置变化实验、损失函数选择实验,验证了改进方案的合理性。

(2)与YOLOv8基本模型、YOLOv7-tiny模型、YOLOv5n系列模型、YOLOv3模型的对比实验表明,改进YOLOv8模型的mAP_0.5值和mAP_0.5:0.95值最高,验证了该模型的优越性。

参考文献

[1]

骆晨,冯玉,吴凯,基于多源停电数据提示学习的电网轻量化停电感知模型[J/OL].现代电力,1-11.(2023-12-18) [2024-05-14].

[2]

LUO Chen, FENG Yu, WU Kai, et al. A lightweight outage perception model for power grids based on prompt learning from multi-source outage data [J/OL].Modern Electric Power,1-11.(2023-12-18) [2024-05-14].

[3]

张美金,郐育,才志君,零序电流分量改进的多判据融合故障选线定位[J].辽宁工程技术大学学报(自然科学版),2020,39(1):71-77.

[4]

ZHANG Meijin, KUAI Yu, CAI Zhijun,et al.Multi-criteria fusion fault line selection for improved zero-sequence current component[J]. Journal of Liaoning Technical University (Natural Science),2020,39 (1):71-77.

[5]

宁鑫,胡馨月,张华,配电线路单相触树接地故障特征分析[J].电力系统及其自动化学报,2023,35(7):137-143.

[6]

NING Xin, HU Xinyue, ZHANG Hua,et al.Characteristic analysis of tree-contact single-phase-to-ground fault in power distribution lines [J]. Proceedings of the CSU-EPSA,2023,35(7):137-143.

[7]

赵燊元,陈天翔,徐会凯,10 kV树线故障树木暂态阻抗变化特性试验研究[J].广西大学学报(自然科学版),2023,48(2):393-406.

[8]

ZHAO Shenyuan, CHEN Tianxiang, XU Huikai,et al Experimental research on transient impedance variation characteristics of 10 kV tree line faulted trees[J].Journal of Guangxi University (Natural Science Edition), 2023,48(2):393-406.

[9]

杨森霖,杨长青,梅吉明,架空输电线路森林火险评估及监测的研究应用现状[J].四川林业科技,2021,42(6):126-130.

[10]

YANG Senlin, YANG Changqing, MEI Jiming,et al.Research and application status on forest fire risk assessment and monitoring for overhead transmission lines[J].Journal of Sichuan Forestry Science and Technology,2021,42(6):126-130.

[11]

赖秋频,杨军,谭本东,基于YOLOv2网络的绝缘子自动识别与缺陷诊断模型[J].中国电力,2019,52(7):31-39.

[12]

LAI Qiupin, YANG Jun, TAN Bendong,et al.An automatic recognition and defect diagnosis model of transmission line insulator based on YOLOv2 network[J].Electric Power,2019,52(7):31-39.

[13]

郝帅,马瑞泽,赵新生,基于卷积块注意模型的YOLOv3输电线路故障检测方法[J].电网技术,2021,45(8):2979-2987.

[14]

HAO Shuai, MA Ruize, ZHAO Xinsheng, et al.Fault detection of YOLOv3 transmission line based on convolutional block attention model[J].Power System Technology,2021,45(8):2979-2987.

[15]

郑伟,杨晓辉,吕中宾,基于改进YOLOv4输电线关键部件实时检测方法[J].科学技术与工程,2021,21(24):10393-10400.

[16]

ZHENG Wei, YANG Xiaohui, Zhongbin LYU,et al.Real-time inspection model for key components of transmission lines based on improved YOLOv4[J].Science Technology and Engineering,2021,21 (24):10393-10400.

[17]

郝帅,杨磊,马旭,基于注意力机制与跨尺度特征融合的YOLOv5输电线路故障检测[J].中国电机工程学报,2023,43(6):2319-2331.

[18]

HAO Shuai, YANG Lei, MA Xu,et al.YOLOv5 transmission line fault detection based on attention mechanism and cross-scale feature fusion[J].Proceedings of the CSEE,2023,43(6):2319-2331.

[19]

程换新,矫立浩,骆晓玲,改进YOLOv8的遥感图像检测算法[J].无线电工程,2024,54(5):1155-1161.

[20]

CHENG Huanxin, JIAO Lihao, LUO Xiaoling,et al.Improved YOLOv8 for remote sensing image detection[J].Radio Engineering,2024,54(5): 1155-1161.

[21]

雷帮军,余翱,余快.基于YOLOv8s改进的小目标检测算法[J].无线电工程,2024,54(4):857-870.

[22]

LEI Bangjun, YU Ao, YU Kuai.Small object detection algorithm based on improved YOLOv8s[J].Radio Engineering,2024,54(4):857-870.

[23]

周飞,郭杜杜,王洋,基于改进YOLOv8的交通监控车辆检测算法[J].计算机工程与应用,2024,60(6):110-120.

[24]

ZHOU Fei, GUO Dudu, WANG Yang,et al.Vehicle detection algorithm based on improved YOLOv8 in traffic surveillance[J].Computer Engineering and Applications,2024,60(6):110-120.

[25]

YU Z F.Sim-YOLO:a real-time Chinese scene text detection method[C]//2023 IEEE 7th Information Technology and Mechatronics Engineering Conference.September 15-17,2023,Chongqing,China. IEEE,2023:2305-2309.

[26]

包从望,朱广勇,邹旺,基于SimAM注意力机制的轴承故障迁移诊断模型[J].机电工程,2024,41(5):862-869, 893.

[27]

BAO Congwang, ZHU Guangyong, ZOU Wang,et al.Rearing fault transfer diagnosis model based on SimAM attention mechanism[J]. Journal of Mechanical & Electrical Engineering,2024,41(5):862-869, 893.

[28]

胡兰兰,邓超.基于SimAM-YOLOv5s的PCB缺陷检测算法[J].无线电工程,2024,54(5):1136-1145.

[29]

HU Lanlan, DENG Chao.PCB defect detection algorithm based on SimAM-YOLOv5s[J].Radio Engineering,2024,54(5):1136-1145.

[30]

田甜,程志友,鞠薇,基于SimAM-ConvNeXt-FL的茶叶病害小样本分类方法研究[J].农业机械学报,2024,55(3):275-281.

[31]

TIAN Tian, CHENG Zhiyou, JU Wei,et al.Small sample classification of tea diseases based on SimAM-ConvNeXt-FL[J].Transactions of the Chinese Society for Agricultural Machinery,2024,55(3):275-281.

[32]

刘向举,刘洋,蒋社想.基于SimAM注意力机制的DCN-YOLOv5水下目标检测[J/OL].重庆工商大学学报(自然科学版),1-9.(2023-10-23)[2024-12-19].

[33]

LIU Xiangju, LIU Yang, JIANG Shexiang. DCN-YOLOv5 underwater target detection based on SimAM attention[J/OL].Journal of Chongqing Technology and Business University (Natural Science Edition),1-9.(2023-10-23)[2024-12-19].

[34]

熊恩杰,张荣芬,刘宇红,面向交通标志的Ghost-YOLOv8检测算法[J].计算机工程与应用,2023,59(20):200-207.

[35]

XIONG Enjie, ZHANG Rongfen, LIU Yuhong,et al.Ghost-YOLOv8 detection algorithm for traffic signs[J].Computer Engineering and Applications,2023,59 (20):200-207.

[36]

王海勇,王志青.基于注意力与特征融合的改进SSD目标检测算法[J].软件,2023,44(4):1-5.

[37]

WANG Haiyong, WANG Zhiqing.Improved SSD object detection algorithm based on attention and feature fusion[J].Software,2023,44 (4):1-5.

[38]

REZATOFIGHI H, TSOI N, GWAK J,et al. Generalized intersection over union: a metric and a loss for bounding box regression[C]//2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition. June 15-20,2019,Long Beach,CA,USA.IEEE,2019:658-666.

[39]

谷长江,高法钦.改进YOLOX-S的金属零件缺陷检测算法研究[J].计算机时代,2023(7):29-33, 37.

[40]

GU Changjiang, GAO Faqin.Improved YOLOX-S metal parts defect detection algorithm [J]. Computer Era,2023(7):29-33, 37.

基金资助

辽宁省教育厅高校基本科研项目(JYTMS20230323)

AI Summary AI Mindmap
PDF (1116KB)

290

访问

0

被引

详细

导航
相关文章

AI思维导图

/