一种残膜回收机防缠绕挑膜装置的制 一种秧草收获机用电力驱动行走机构

绝缘子缺陷检测方法、系统、设备及存储介质与流程

2023-01-16 22:52:28 来源:中国专利 TAG:


1.本发明设计电力技术领域,尤其涉及一种绝缘子缺陷检测方法、系统、设备及存储介质。


背景技术:

2.绝缘子是一种特殊的绝缘控件,它处于电线杆塔与导线承接部,变电所构架与线路连接处,能够在架空输电线路中起到支撑导线、防止电流接地的双重作用。由于我国输电线路分布十分广泛,地理环境复杂,输电线缆长期暴露在野外,绝缘子可能会受到雷击,材料老化等而产生一系列的生锈腐蚀、破损、过电压击穿等问题,根据需求我们把绝缘子异常分为三类分别为雷击(放电)、破损、匝丝捆绑异常。分析绝缘子常见故障,主要是为了防止由于环境和电负荷条件发生变化引起的各种机电应力导致绝缘子绝缘失效,从而损害电力线路的使用和运行寿命。
3.而这些缺陷通过人工检测会有效率低,检出率低,安全性低等问题。传统的分类任务会将输入图片分配到某一类。而多标签分类任务不同于一般的分类任务,多标签分类可以分解为若干个独立的二分类问题,输入图片可能被分配到所有的类,也可能不分给任何一类。以绝缘子举例:一个绝缘子可能同时存在雷击(破损)、匝丝捆绑异常、脏污三种异常,也可能这三种异常同时不存在,那么这就是一个正常的绝缘子。但是,绝缘子多标签分类任务中的难点就是正负样本严重不均衡,存在缺陷的绝缘子相对于正常绝缘子非常少,相关技术并未考虑正负样本的均衡性,尽管通过机器学习可以实现故障的检测,但是在正负样本量极度不均衡的情况下,使得训练模型的学习能力较弱,从而得到的检测模型检测的精度较差。


技术实现要素:

4.本发明的目的在于提出一种绝缘子缺陷检测方法、系统、设备及存储介质,通过更新梯度比更新损失函数loss,再根据上一个训练周期的损失函数更新多标签分类模型,能够解决正负样本不均衡的问题,提高多标签分类模型的学习能力,从而提高多标签分类模型的检测精度,及时判断供电线路中绝缘子存在的各种缺陷隐患。
5.为达到上述目的,本发明第一方面实施例提出了一种绝缘子缺陷检测方法,包括以下步骤:获取待检测图像;将所述待检测图像输入预先训练好的绝缘子目标检测模型,输出第一检测结果;若所述第一检测结果中包括绝缘子图像,则对所述绝缘子图像按照第一预设尺寸进行第一裁剪;将进行第一裁剪后的绝缘子图像输入预先训练好的多标签分类模型,输出第二检测结果,其中,所述第二检测结果包括绝缘子的属性分类,其中,所述多标签分类模型的训练过程如下:获取第一训练集,其中,所述第一训练集中的各图像具有第一标签,所述第一标签包括绝缘子目标框的中心点、宽度和高度;对所述第一训练集中的各图像按照第二预设尺寸进行第二裁剪,得到第二训练集,其中,所述第二训练集中的各图像为绝缘子图像;对所
述第二训练集中的图像进行标注,得到第二标签,其中,所述第二标签包括绝缘子的属性分类;构建多标签分类模型;对所述第二训练集中的各图像进行随机裁剪与水平翻转和/或上下翻转的数据增强,将增强后的图像输入所述多标签分类模型,输出相应的属性分类结果;根据所述属性分类结果和对应的第二标签构建损失函数,根据所述损失函数训练所述多标签分类模型。
6.根据本发明实施例的绝缘子缺陷检测方法,能够解决正负样本不均衡的问题,提高多标签分类模型的学习能力,从而提高多标签分类模型的检测精度,及时判断供电线路中绝缘子存在的各种缺陷隐患。
7.另外,本发明实施例的绝缘子缺陷检测方法,还具有如下附加的技术特征:根据本发明的一个实施例,所述绝缘子目标检测模型的训练过程如下:构建绝缘子目标检测模型;利用所述第一训练集中图像训练所述绝缘子目标检测模型。
8.根据本发明的一个实施例,所述第一裁剪后的绝缘子图像中绝缘子的宽度与所述第一裁剪前的绝缘子图像中相应绝缘子的宽度的比值为1.111,所述第一裁剪后的绝缘子图像中绝缘子的高度与所述第一裁剪前的绝缘子图像中相应绝缘子的高度的比值为1.111;所述第二裁剪后的图像中绝缘子的宽度与所述第二裁剪前的图像中相应绝缘子的宽度的比值为1.222,所述第二裁剪后的图像中绝缘子的高度与所述第二裁剪前的图像中相应绝缘子的高度的比值为1.222。
9.根据本发明的一个实施例,通过下式构建损失函数loss:loss=bce(prob,target)*weight,weight=w
pos
*target w
neg
*(1-target),w
pos
=1 4(1-w
neg
),w
neg
=f(g
t-1
),,其中,prob为所述属性分类结果,target为所述第二标签,w
pos
为正梯度重加权权重,w
neg
为负梯度重加权权重,f(x)为映射函数,g
t-1
为上一个训练周期的梯度比。
10.根据本发明的一个实施例,通过下式得到更新后的梯度比g
t
:,greweight
pos
(loss)=w
pos
*g
pos
(loss),greweight
neg
(loss)=w
neg
*g
neg
(loss),g
pos
(loss)=gard*target,g
neg
(loss)=gard*(1-target),gard=target*(prob-1) (1-target)*prob,其中,greweight
pos
(loss)为重加权后的正样本梯度,greweight
neg
(loss)为重加权后的负样本梯度,g
pos
(loss)为正样本梯度,g
neg
(loss)为负样本梯度,grad为原始梯度。
11.根据本发明的一个实施例,所述根据所述损失函数训练所述多标签分类模型,包括:在当前训练周期更新梯度比后,逐渐平衡正负样本对所述多标签分类模型的影响,保存
测试指标最高的多标签分类模型。
12.根据本发明的一个实施例,所述方法还包括:根据所述第二检测结果将存在缺陷的绝缘子在所述待检测图像中示出。
13.为达到上述目的,本发明第二方面实施例提出了一种计算机可读存储介质,其上存储有计算机程序,所述计算机程序被处理器执行时,实现所述的方法。
14.为达到上述目的,本发明第三方面实施例提出了一种绝缘子缺陷检测设备,包括存储器、处理器以及存储在所述存储器上的计算机程序,所述计算机程序被所述处理器执行时,实现所述的方法。
15.为达到上述目的,本发明第四方面实施例提出了一种绝缘子缺陷检测系统,包括:图像采集单元,用于采集待检测图像;绝缘子目标检测单元,用于将所述待检测图像输入预先训练好的绝缘子目标检测模型,输出第一检测结果;裁剪单元,用于在所述第一检测结果中包括绝缘子图像时,对所述绝缘子图像按照第一预设尺寸进行第一裁剪;多标签分类单元,用于将进行第一裁剪后的绝缘子图像输入预先训练好的多标签分类模型,输出第二检测结果,其中,所述第二检测结果包括绝缘子的属性分类;其中,所述多标签分类模型的训练过程如下:获取第一训练集,其中,所述第一训练集中的各图像具有第一标签,所述第一标签包括绝缘子目标框的中心点、宽度和高度;对所述第一训练集中的各图像按照第二预设尺寸进行第二裁剪,得到第二训练集,其中,所述第二训练集中的各图像为绝缘子图像;对所述第二训练集中的图像进行标注,得到第二标签,其中,所述第二标签包括绝缘子的属性分类;构建多标签分类模型;对所述第二训练集中的各图像进行随机裁剪与水平翻转和/或上下翻转的数据增强,将增强后的图像输入所述多标签分类模型,输出相应的属性分类结果;根据所述属性分类结果和对应的第二标签构建损失函数,根据所述损失函数训练所述多标签分类模型。
16.本发明附加的方面和优点将在下面的描述中部分给出,部分将从下面的描述中变得明显,或通过本发明的实践了解到。
附图说明
17.图1是本发明一个实施例的绝缘子缺陷检测方法的流程示意图;图2是本发明一个实施例的绝缘子目标检测模型的训练流程示意图;图3是本发明一个实施例的多标签分类模型的训练流程示意图;图4是本发明一个具体实施例的对绝缘子图像按照第一预设尺寸进行第一裁剪的示意图;图5是本发明一个具体实施例的对第一训练集中的各图像按照第二预设尺寸进行第二裁剪的示意图;图6是本发明一个具体实施例的对第二训练集中的各图像进行随机裁剪与水平翻转和/或上下翻转的数据增强的示意图;图7是本发明一个实施例的绝缘子缺陷检测系统的结构示意图。
具体实施方式
18.下面详细描述本发明的实施例,所述实施例的示例在附图中示出,其中自始至终相同或类似的标号表示相同或类似的元件或具有相同或类似功能的元件。下面通过参考附图描述的实施例是示例性的,旨在用于解释本发明,而不能理解为对本发明的限制。
19.下面参考附图描述本发明实施例的绝缘子缺陷检测方法、系统、设备及存储介质。
20.图1是本发明一个实施例的绝缘子缺陷检测方法的流程示意图。如图1所示,绝缘子缺陷检测方法,包括以下步骤:s101、获取待检测图像。
21.具体地,可通过将高分辨率摄像机搭载在无人机上,获取待检测图像。
22.s102、将待检测图像输入预先训练好的绝缘子目标检测模型,输出第一检测结果。
23.作为一个示例,参见图2,绝缘子目标检测模型的训练过程包括步骤s201-s202:s201、构建绝缘子目标检测模型。
24.具体地,因为搭载高分辨率摄像机的无人机采集的第一训练集中的图像分辨率较高,所以采用1280*1280为输入的yolov5s目标检测算法,本发明的绝缘子目标检测模型采用的yolov5s算法开源的yolov5,最终在测试集上的绝缘子测试ap(ap在目标检测中是一种常见的指标,其结合了准确率与召回率)值为0.978。yolov5s目标检测算法原理:输入图像经过多层的卷积网络的特征提取,然后经过多尺度的输出层输出,通过预设的锚框对目标进行回归计算,通过置信度判断候选框中有无目标,yolov5算法最终输出的结果经过非极大值抑制操作,得到最终的检测框。由此,通过绝缘子目标检测模型,找到图像中感兴趣的物体。
25.s202、利用第一训练集中图像训练绝缘子目标检测模型。
26.具体地,利用下述第一训练集中图像训练绝缘子目标检测模型,得到预先训练好的绝缘子目标检测模型。
27.进一步地,将步骤s101中获取的待检测图像输入预先训练好的绝缘子目标检测模型,输出第一检测结果,即从待检测图像中裁剪出所有绝缘子。
28.s103、若第一检测结果中包括绝缘子图像,则对绝缘子图像按照第一预设尺寸进行第一裁剪。
29.具体地,对绝缘子图像按照第一预设尺寸进行第一裁剪,可包括:获取绝缘子图像中所有绝缘子所占像素面积的宽度和高度、绝缘子图像中一绝缘子的宽度和高度以及第一预设尺寸;计算绝缘子图像中一绝缘子的宽度和第一预设尺寸中宽度的第一乘积,并计算第一乘积与绝缘子图像中所有绝缘子所占像素面积的宽度的比值,得到第一扩张的宽度;计算绝缘子图像中一绝缘子的高度和第一预设尺寸中高度的第二乘积,并计算第二乘积与绝缘子图像中所有绝缘子所占像素面积的高度的比值,得到第一扩张的高度;根据第一扩张的宽度和第一扩张的高度,对绝缘子图像中一绝缘子进行第一扩张;根据第一预设尺寸,对第一扩张后的绝缘子图像中相应绝缘子进行第一裁剪。由此,对绝缘子图像按照第一预设尺寸进行第一裁剪,能够使得绝缘子整体显示于绝缘子图像中。
30.例如:如图4所示,设绝缘子图像中所有绝缘子所占像素面积的高度和宽度(288,288),绝缘子图像中一绝缘子(参见图4中的圆形)的宽度、高度分别为w,h,第一预设尺寸为(320,320),根据上述方案可得公式w/288 = w1/320,h/288=h1/320,推导出第一扩张的宽
度和高度w1=1.111w,h1=1.111h,根据第一扩张的宽度和第一扩张的高度,对绝缘子图像中一绝缘子进行第一扩张;根据第一预设尺寸,对第一扩张后的绝缘子图像中相应绝缘子进行第一裁剪。也就是说,第一裁剪后的绝缘子图像中绝缘子的宽度为第一裁剪前的绝缘子图像中相应绝缘子的宽度的1.111倍,第一裁剪后的绝缘子图像中绝缘子的高度也为第一裁剪前的绝缘子图像中相应绝缘子的高度的1.111倍。
31.s104、将进行第一裁剪后的绝缘子图像输入预先训练好的多标签分类模型,输出第二检测结果,其中,第二检测结果包括绝缘子的属性分类。
32.作为一个示例,参见图3,多标签分类模型的训练过程包括步骤s301-s306:s301、获取第一训练集,其中,第一训练集中的各图像具有第一标签,第一标签包括绝缘子目标框的中心点、宽度和高度。
33.具体地,使用labelme工具对搭载高分辨率摄像机的无人机采集的图像进行对角标注,即在左上角、右下角进行标注或右上角、左下角进行标注,得到标注框,再将同一图像中,标注框的坐标除以图像的宽度值、高度值,即进行归一化处理,经过归一化处理后,生成第一标签,其中,第一标签的格式可以为:绝缘子的有无分类,图像中心点、图像的高度和宽度,即(class,centerx,centery,width,height)。
34.s302、对第一训练集中的各图像按照第二预设尺寸进行第二裁剪,得到第二训练集,其中,第二训练集中的各图像为绝缘子图像。
35.具体地,对第一训练集中的各图像按照第二预设尺寸进行第二裁剪,可包括:获取图像中所有绝缘子所占像素面积的宽度和高度、图像中一绝缘子的宽度和高度以及第二预设尺寸;计算图像中一绝缘子的宽度和第二预设尺寸中宽度的第三乘积,并计算第三乘积与图像中所有绝缘子所占像素面积的宽度的比值,得到第二扩张的宽度;计算图像中一绝缘子的高度和第二预设尺寸中高度的第四乘积,并计算第四乘积与图像中所有绝缘子所占像素面积的高度的比值,得到第二扩张的高度;根据第二扩张的宽度和第二扩张的高度,对图像进行第二扩张;根据第二预设尺寸,对第二扩张后的图像中相应绝缘子进行第二裁剪。由此,通过对第一训练集中的各图像按照第二预设尺寸进行第二裁剪,能够使得绝缘子整体显示于图像中。
36.例如:如图5所示,设第一训练集中的一图像中所有绝缘子所占像素面积的高度和宽度为(288,288),该图像中一绝缘子(参见图5中的圆形)的宽度、高度分别为w,h,第二预设尺寸为(352,352),根据上述方案可得公式w/288=w2/352,h/288=h2/352,推导出第二扩张的宽度和高度w2= 1.222w,h2=1.222h,根据第二扩张的宽度和第二扩张的高度,对图像进行第二扩张;根据第二预设尺寸,对第二扩张后的图像中相应绝缘子进行第二裁剪。也就是说,第二裁剪后的图像中绝缘子的宽度为第二裁剪前的图像中相应绝缘子的宽度的1.222倍,第二裁剪后的图像中绝缘子的高度也为第二裁剪前的图像中相应绝缘子的高度的1.222倍。
37.s303、对第二训练集中的图像进行标注,得到第二标签,其中,第二标签包括绝缘子的属性分类。
38.具体地,使用labelbee工具对第二训练集中的图像进行属性分类标注,绝缘子的属性分类包括正常、异常,异常包括发生雷击(破损)、匝丝捆绑异常、脏污中的至少一者。也就是说,对于每个绝缘子来说这三种异常属性可以同时满足多项,也可以只满足其中某一
项。
39.s304、构建多标签分类模型。
40.s305、对第二训练集中的各图像进行随机裁剪与水平翻转和/或上下翻转的数据增强,将增强后的图像输入多标签分类模型,输出相应的属性分类结果。
41.其中,对第二训练集中的图像进行随机裁剪与水平翻转和/或上下翻转的数据增强,能够消除位置差异,增加数据样本数量。举例而言,如图6所示,设第二训练集中的图像的宽度和高度为(352,352),预设宽高比例为(0.865,0.935),对第二训练集中的图像按照预设宽高比例(0.865,0.935)进行随机裁剪,将绝缘子图像的宽高设置为(320,320),最后对随机裁剪后的绝缘子图像进行水平翻转和/或上下翻转。
42.需要说明的是,绝缘子目标检测模型存在检测误差(如由于检测精度低于人工标注精度,检测框定位不够精准),而多标签分类模型训练时,训练数据由人工标注,训练数据不存在绝缘子目标检测模型输出的异常情况,导致绝缘子目标检测模型的检测结果与多标签分类模型训练时输入图像的区域不完全一致,影响模型最终效果,所以本实施例中采用随机裁剪与水平翻转和/或上下翻转的数据增强将区域保持一致。
43.s306、根据属性分类结果和对应的第二标签构建损失函数,根据损失函数训练多标签分类模型。
44.具体地,通过下式构建损失函数loss:loss=bce(prob,target)*weight,weight=w
pos
*target w
neg
*(1-target),w
pos
=1 4(1-w
neg
),w
neg
=f(g
t-1
),,其中,prob为属性分类结果,target为第二标签,w
pos
为正梯度重加权权重,w
neg
为负梯度重加权权重,f(x)为映射函数,g
t-1
为上一个训练周期的梯度比。
45.需要说明的是,在第一个训练周期中不存在g
t-1
,所以也不存在w
pos
、w
neg
,初始化g
t-1
=w
pos
=w
neg
=1,计算损失函数loss:loss=bce(prob,target)*weight,其中,weight=w
pos
*target w
neg
*(1-target)。
46.另外,映射函数f(x)的原理为:刚开始训练时,因为正样本(即异常绝缘子)数据量远小于负样本(即正常绝缘子)数据量,所以g
t-1
在开始训练时接近于0,又因为f(x)是一个类似于sigmoid的函数,当传入的g
t-1
小于0.8时,12(x-0.8)这项小于0,所以f(g
t-1
)=w
neg
﹤0.5,w
pos
﹥1,每轮迭代后都会减少负样本梯度,增加正样本梯度,梯度比慢慢趋近于1达到平衡的效果。
47.进一步地,根据损失函数训练多标签分类模型,可包括:在当前训练周期更新梯度比后,逐渐平衡正负样本对所述多标签分类模型的影响,保存测试指标最高的多标签分类模型。
48.需要说明的是,在进行绝缘子异常分类任务时,存在两个问题:第一,正常绝缘子占绝大多数,正常绝缘子与异常绝缘子占比严重不平衡;第二,在异常绝缘子中不同分类的异常绝缘子占比不平衡。例如一共存在10w个绝缘子数据其中只有5000个绝缘子存在异常,
在5000个异常样本中雷击(破损)、匝丝捆绑异常、脏污分别占500,1500,4000,同时还存在一个绝缘子存在多种异常情况。对于这种不平衡的数据集,常见的做法是给正负样本所产生的loss加上与数量成反比的权重,同时给不同寻常分类也加上与数量成反比的权重。但是如果仅仅用正负样本数量来反映模型的训练情况,依旧存在一定的缺陷,因为数据集中大量负样本即正常绝缘子都是简单样本,这些简单样本产生的loss已经接近于0,模型已经能预测的很好,模型已经无法从中学到知识。同样的,在正样本即异常绝缘子中也可能存在特征重叠的情况,比如有两张重复的图片,这些重复的正样本也没有带来给模型额外的知识,所以只是单独使用数量比来训练模型这不是最好的,需要考虑不同样本因为难易程度对模型造成的影响。本实施例中使用梯度来反映样本对模型造成的影响,当样本产生的loss越大导致梯度越大代表样本较难,当样本产生的loss越小导致梯度越小代表样本较简单。
49.为了解决正负样本不均衡对训练的影响,本实施例中利用重加权后的正样本梯度和重加权后的负样本梯度,使得梯度比随着训练的进行由初始的不均衡数值(例如0.2)优化至趋近于1。
50.具体地,通过下式得到更新后的梯度比g
t
:,greweight
pos
(loss)=w
pos
*g
pos
(loss),greweight
neg
(loss)=w
neg
*g
neg
(loss),g
pos
(loss)=gard*target,g
neg
(loss)=gard*(1-target),gard=target*(prob-1) (1-target)*prob,其中,greweight
pos
(loss)为重加权后的正样本梯度,greweight
neg
(loss)为重加权后的负样本梯度,g
pos
(loss)为正样本梯度,g
neg
(loss)为负样本梯度,grad为原始梯度。
51.具体而言,本发明以加权平均的方式更新梯度比,即根据公式求出g
t
,并求出下个训练周期需要的w
pos
、w
neg
,最后判断更新后的梯度比g
t
与上一个训练周期的梯度比g
t-1
是否相等,若不相等,则使用上一个训练周期的损失函数更新多标签分类模型,并进行下一个训练周期的训练。并在当前训练周期更新梯度比后,逐渐平衡正负样本对所述多标签分类模型的影响,保存测试指标最高的多标签分类模型,其中,测试指标为precision(精确度)和recall(召回率),最优模型(即预先训练好的多标签分类模型)的测试指标为:雷击(破损)precision:0.824,recall:0.781;匝丝绑扎异常precision:0.854,recall:0.882;脏污precision:0.954,recall:0.946。
52.需要说明的是,训练时会保留第一训练集中10%的数据作为测试集,测试集中的图片不进行训练,只用来计算测试指标。具体地,将测试集中的图像输入当前更新后的多标签分类模型,得到测试结果;根据测试结果计算精确度以及召回率,得到测试指标。
53.进一步地,将步骤s103中裁剪成预设尺寸的绝缘子图像输入预先训练好的多标签
分类模型,输出第二检测结果,其中,第二检测结果包括绝缘子的属性分类。由此,通过多标签分类模型,判断绝缘子是否存在异常及绝缘子的异常类型。
54.需要说明的是,绝缘子缺陷检测方法还可包括:根据第二检测结果将存在缺陷的绝缘子在待检测图像中示出。
55.综上所述,该绝缘子缺陷检测方法,通过获取待检测图像;将待检测图像输入预先训练好的绝缘子目标检测模型,输出第一检测结果;若第一检测结果中包括绝缘子图像,则将绝缘子图像裁剪至预设尺寸;将裁剪后的绝缘子图像输入预先训练好的多标签分类模型,输出第二检测结果,其中,通过更新梯度比更新损失函数loss,再根据上一个训练周期的损失函数更新多标签分类模型,能够解决正负样本不均衡的问题,提高多标签分类模型的学习能力,从而提高多标签分类模型的检测精度,及时判断供电线路中绝缘子存在的各种缺陷隐患。
56.本发明还提供了一种计算机可读存储介质,其上存储有计算机程序,计算机程序被处理器执行时,实现上述实施例的绝缘子缺陷检测方法。
57.本发明实施例的计算机可读存储介质,在其上存储的与上述绝缘子缺陷检测方法对应的计算机程序被处理器执行时,能够提高多标签分类模型的检测精度,及时判断供电线路中绝缘子存在的各种缺陷隐患。
58.本发明还提供了绝缘子缺陷检测设备,包括存储器、处理器以及存储在存储器上的计算机程序,计算机程序被处理器执行时,实现上述实施例的绝缘子缺陷检测方法。
59.本发明实施例的绝缘子缺陷检测设备,在其存储器上存储的与上述绝缘子缺陷检测方法对应的计算机程序被处理,能够提高多标签分类模型的检测精度,及时判断供电线路中绝缘子存在的各种缺陷隐患。
60.图7是本发明一个实施例的绝缘子缺陷检测系统的结构示意图。如图7所示,绝缘子缺陷检测系统100,应用上述的绝缘子缺陷检测方法,包括:图像采集单元10,绝缘子目标检测单元20,裁剪单元30和多标签分类单元40。
61.其中,图像采集单元10,用于采集待检测图像;绝缘子目标检测单元20,用于将待检测图像输入预先训练好的绝缘子目标检测模型,输出第一检测结果;裁剪单元30,用于在第一检测结果中包括绝缘子图像时,对绝缘子图像按照第一预设尺寸进行第一裁剪;多标签分类单元40,用于将进行第一裁剪后的绝缘子图像输入预先训练好的多标签分类模型,输出第二检测结果,其中,第二检测结果包括绝缘子的属性分类。
62.在该实施例中,多标签分类模型的训练过程如下:获取第一训练集,其中,第一训练集中的各图像具有第一标签,第一标签包括绝缘子目标框的中心点、宽度和高度;对第一训练集中的各图像按照第二预设尺寸进行第二裁剪,得到第二训练集,其中,第二训练集中的各图像为绝缘子图像;对第二训练集中的图像进行标注,得到第二标签,其中,第二标签包括绝缘子的属性分类;构建多标签分类模型;对第二训练集中的各图像进行随机裁剪与水平翻转和/或上下翻转的数据增强,将增强后的图像输入多标签分类模型,输出相应的属性分类结果;根据属性分类结果和对应的第二标签构建损失函数,根据损失函数训练多标签分类模型。
63.具体地,通过下式构建损失函数loss:loss=bce(prob,target)*weight,
weight=w
pos
*target w
neg
*(1-target),w
pos
=1 4(1-w
neg
),w
neg
=f(g
t-1
),,其中,prob为属性分类结果,target为第二标签,w
pos
为正梯度重加权权重,w
neg
为负梯度重加权权重,f(x)为映射函数,g
t-1
为上一个训练周期的梯度比。
64.通过下式得到更新后的梯度比g
t
:,greweight
pos
(loss)=w
pos
*g
pos
(loss),greweight
neg
(loss)=w
neg
*g
neg
(loss),g
pos
(loss)=gard*target,g
neg
(loss)=gard*(1-target),gard=target*(prob-1) (1-target)*prob,其中,greweight
pos
(loss)为重加权后的正样本梯度,greweight
neg
(loss)为重加权后的负样本梯度,g
pos
(loss)为正样本梯度,g
neg
(loss)为负样本梯度,grad为原始梯度。
65.进一步地,在当前训练周期更新梯度比后,逐渐平衡正负样本对所述多标签分类模型的影响,保存测试指标最高的多标签分类模型。
66.需要说明的是,本发明实施例的绝缘子缺陷检测系统的其他具体实施方式可参见本发明上述实施例的绝缘子缺陷检测方法的具体实施方式。
67.综上所述,该绝缘子缺陷检测系统,通过更新梯度比更新损失函数loss,再根据上一个训练周期的损失函数更新多标签分类模型,能够解决正负样本不均衡的问题,提高多标签分类模型的学习能力,从而提高多标签分类模型的检测精度,及时判断供电线路中绝缘子存在的各种缺陷隐患。
68.尽管上面已经示出和描述了本发明的实施例,可以理解的是,上述实施例是示例性的,不能理解为对本发明的限制,本领域的普通技术人员在本发明的范围内可以对上述实施例进行变化、修改、替换和变型。
再多了解一些

本文用于创业者技术爱好者查询,仅供学习研究,如用于商业用途,请联系技术所有人。

发表评论 共有条评论
用户名: 密码:
验证码: 匿名发表

相关文献