2. 中国南方电网超高压输电公司广州局,广东 广州 510000
2. China Southern Power Grid EHV Transmission Company Guangzhou Bureau, Guangzhou 510000, China
随着计算机视觉和无人机技术的快速发展,无人机和计算机视觉技术相结合在智能电网中的应用需求越来越多,而且一些成果已经应用到输电线路电力部件的智能巡检和在线监测中[1]。实现对输电线路电力部件智能巡检的关键是对电力部件进行识别和定位。输电线路的主要部件包括绝缘子、间隔棒、防震锤和均衡环等,在无人机拍摄电力部件的过程中,由于距离的因素,这些电力部件距离无人机较远,在图像上显示的尺寸较小,是典型的小目标识别问题。
传统图像识别方法以图像二值化和边缘检测为基础,通过尺度不变特征变换[2]、局部二值模式[3]等算法来提取图像的梯度、颜色、纹理、边缘和形态等显著特征[4],但这些特征提取算法需要根据目标预先设定特征参数,对拍摄角度距离差异、光照影响和复杂背景干扰较为敏感,难以设计通用的特征提取模型来描述不同情景下的物体。因此,其识别准确率较低、鲁棒性较差,不适用于电力小部件识别。
随着深度学习的迅速发展以及硬件水平的提高,深度卷积神经网络在计算机视觉领域取得了巨大的成功,而且电力部件的数据集越来越多,为深度学习在电力部件识别和定位中的应用奠定了基础。文献[5]通过优化更快的基于区域建议卷积神经网络(faster region-based convolutional neural network, Faster RCNN)[6]损失函数和自建ImageNet图像样本库对输电线路多目标图像进行识别和定位,但改进后的网络模型依然保留高层特征,抛弃浅层特征,对小目标识别和定位效果较差,且检测速度较慢,实时性难以得到保证,无法进行实际应用。基于回归的one-stage目标检测算法以SSD[7]、YOLO[8]为主,one-stage算法在检测速度方面远快于基于区域兴趣的two-stage目标检测算法,能够达到实时检测的要求。YOLO算法将特征图等分为n×n的网格,但是当多个小目标落在一个网格中的时候,会漏检和误检目标。利用SSD检测模型检测小目标时,其只用VGG16特征提取网络结构中的conv4_3层,语义信息不够丰富,因此SSD模型检测小目标效果较差。Liu 等[9]对SSD进行了改进,充分利用特征层之间的上下文信息,使用残差网络提升特征网络的表征能力,提出了引入反卷积的单阶段多框检测器(deconvolutional single shot detector, DSSD)模型[10],但检测速度较慢。
针对以上问题,本文提出了基于注意力机制和多尺度特征融合的amSSD电力部件识别网络,在经典SSD网络的特征提取阶段引入注意力机制,进一步提升网络的特征提取能力,增强特征图的语义信息。同时,为解决SSD识别尺寸较小电力部件效果差的问题,使用膨胀卷积操作来增大浅层特征图的感受野,并通过反卷积操作将不同语义信息的特征图进行融合,提升算法的检测精度,而且有较高的检测效率。利用实际飞行采集的电力部件数据对算法进行了测试验证,并与SSD算法进行对比分析。
1 相关技术理论和算法 1.1 注意力机制在计算机视觉领域,注意力机制包括基于项的注意力[11]和基于位置的注意力[12]。基于项和基于位置注意力机制的输入形式不同,基于项注意力的输入可以是特征图、向量或者矩阵形式的序列项,或者添加预处理操作来生成序列项;而基于位置注意力的输入形式为单独的特征图,可以通过位置指定所有的目标。基于位置的注意力直接作用在特征图的某一区域上,应用较为广泛;基于项的注意力能够进行端对端训练,并且不会较大地改动原有结构。因此,本文在特征提取层引入基于项的通道注意力机制网络结构(SENet)[13]。
1.2 反卷积小目标往往更依赖浅层特征,因为浅层特征有更高的分辨率,浅层包含的细节信息比较多。因此,本文采用反卷积操作[14],将具有高语义信息的高层,经过反卷积操作后,与浅层特征图相融合,使浅层特征图语义信息更加充足,对小目标检测更有效。反卷积,也即卷积操作的逆运算,通过转置卷积核,与卷积后的结果再次进行卷积,卷积过程与反卷积过程示意图如图1所示。
Download:
|
|
图1(a)为卷积正向传播操作,5像素×5像素的特征图经过3×3的卷积核卷积运算,步长为2,填充为1的卷积操作之后,得到大小为3像素×3像素的特征图;图1(b)为反卷积操作,2像素×2像素的特征图通过填充补零与3×3的卷积核进行卷积运算,得到4像素×4像素的特征图。反卷积的输入输出关系为
$o' = s(i' - 1) + k - 2p$ |
式中:
SSD模型以VGG16为基础网络,但是将VGG16的全连接层fc6和fc7转换成卷积层conv6和conv7,conv6层和conv7层的卷积核大小分别为3×3和1×1,并添加特征提取层。VGG16中的conv4_3层特征图大小是38像素×38像素,主要用于识别小目标。使用SSD网络中6种不同的特征层组成多尺度特征层,实现多尺度检测。SSD网络结构如图2所示。
Download:
|
|
SSD检测算法使用的6种特征图大小分别为38像素×38像素、19像素×19像素、10像素×10像素、5像素×5像素、3像素×3像素和1像素×1像素,针对不同的特征图,都设置了不同大小的默认框,设置默认框的计算为
${S_k} = {S_{\min }} + \frac{{{S_{\max }} - {S_{\min }}}}{{m - 1}}(k - 1),k = \{ 1,2,\cdots,m\} $ |
式中:k表示第k个特征图;
${W_k} = {S_k}\sqrt {{r_n}} ,n = \{ 1,2,3,4,5\} $ |
${H_k} = \frac{{{S_k}}}{{\sqrt {{r_n}} }},n = \{ 1,2,3,4,5\} $ |
当r=1时,增加默认框的尺度为
$W_k^6 = H_k^6 = \sqrt {{S_k}{S_{k + 1}}} $ |
通过上述方法,SSD算法可获得大量的正负样本,算法通过比较交并比(intersection-over-union, IOU)值,得到真实目标框。实际中,默认框中很小一部分为真实目标框,为使正负样本保持平衡,SSD 算法采用困难样本挖掘方法[15],随机抽取负样本,按照置信度分数从高到低排列,将误差小的作为正样本,并使正负样本数量的比例为1:3。
1.3.3 损失函数SSD算法的目标损失函数包括分类预测损失和位置回归损失两部分。总体目标损失函数为
$L(x,c,l,g) = \frac{1}{N}({L_{{\rm{conf}}}}(x,c) + \alpha {L_{{\rm{loc}}}}(x,l,g))$ |
式中:N为匹配的默认框数目,当N为0时,损失设为0;α为平衡分类预测损失和位置回归损失的权重参数;Lloc为位置回归损失。使用Smooth L1损失函数[16]计算预测的偏移量(l)与真实的偏移量(g)之间的损失,然后对默认框的位置偏移量进行回归计算:
$\begin{array}{l} {L_{{\rm{loc}}}}(x,l,g) = \\ \displaystyle\sum\nolimits_{i \in {\rm{pos}}}^N {\displaystyle\sum\nolimits_{m \in (x,y,w,h)} {x_{ij}^k{\rm{smoot}}{{\rm{h}}_{_{L1}}}(l_i^m - \hat g_j^m} } ) \end{array} $ |
式中:pos为正样本默认框;li为预测的第i个正样本的位置偏移量;
$\hat g_j^{cx} = (g_j^{cx} - d_i^{cx})/d_i^w,\hat g_j^{cy} = (g_j^{cy} - d_i^{cy})/d_i^h$ |
$\hat g_j^w = {\rm{log}}(g_j^w/d_i^w),\hat g_j^h = {\rm{log}}(g_j^h/d_i^h)$ |
Lconf为分类预测损失,使用Softmax损失函数:
${L_{{\rm{conf}}}}(x,c) = - {\sum\nolimits_{i \in {\rm{pos}}}^N {{x_{ij}}{\rm{log}}(\hat c} _i}) - \sum\nolimits_{i \in {\rm{neg}}} {{\rm{log}}(} \hat c_i^0)$ |
式中:neg为负样本检测框;
本文提出注意力机制和反卷积多尺度融合的输电线路电力部件识别模型,通过膨胀卷积、通道加权更新、反卷积等操作,增强浅层特征层的感受野和语义信息,提升模型对较小尺寸电力部件的识别能力。改进后的模型amSSD如图3所示。
Download:
|
|
SENet结构能很容易添加到常见的卷积神经网络结构中,增强特征提取阶段的感受野,提高与目标相关特征通道的权重,抑制与目标无关的特征通道,进一步提升特征图的语义信息。在SENet结构中,注意力机制的最重要的操作是挤压(squeeze)和激励(excitation)。首先,对输入SENet结构中的特征图进行全局平均池化(global average pooling,GAP)[17],得到长度为M的实数列,使每个通道上的特征图具有全局的感受野,从而使感受野较小的浅层特征图能够利用全局信息[18],提升网络的特征提取能力,得到图像更为丰富的语义信息;其次,将长度为M的实数列输入到全连接层,先降维成
Download:
|
|
由于尺寸较小的电力部件主要通过Conv4_3层特征图进行识别,因此需要提升该层的语义信息。本文采取的改进措施主要有以下2个方面:
1) 对细节信息较高的浅层Conv3_1层做膨胀卷积,目的是增大感受野,损失较少的细节信息,使得特征图具有更多的全局信息。本文选取的dilation值为2、步长为2、边缘填充值为2;再将膨胀卷积后的特征图与Conv4_3以特征通道拼接的方式进行融合,得到38像素×38像素×1024像素大小的特征图;最后采用1×1×512大小的卷积核对融合后的特征图执行卷积操作,输出38像素×38像素×512像素大小的语义信息更加丰富的特征图Conv4_d,有利于识别尺寸较小的电力部件。
2) 对语义信息较丰富的高层特征图经过反卷积操作与浅层特征图融合,从而使浅层特征图能够融入较丰富的语义信息,然后进行多尺度特征图识别图像中的电力部件。首先,采用反卷积操作扩充 Conv11_2层的特征图,和Conv10_2层特征图大小相同;然后,将扩充后的特征图与Conv10_2特征图以对应通道相加的方式进行融合,更新Conv10_2层的特征图;最后,将更新后的Conv10_2层的特征图采取同样的方式进行反卷积扩充,并与Conv9_2层特征图融合,并更新Conv9_2层的特征图。同理,依次更新Conv8_2、fc7和Conv4_3层的特征图。通过以上2种改进措施,得到的Conv4_3具有较强的细节分辨率信息和较高的语义感受野,能够较好地进行较小尺寸电力部件识别和定位。
最后在预测模块中对更新后的多尺度特征图进行分类和回归计算,得到电力部件识别定位结果。
3 实验及结果分析本文实验使用Ubuntu 16.04操作系统搭建深度学习环境,显卡型号为NVIDIA GeForce GTX 2080Ti,采用的深度学习框架为Pytorch 1.3.1。
3.1 数据集及实验参数本文的数据集为航天图景(北京)科技有限公司无人机拍摄的输电线路电力部件图像,包括绝缘子(insulator)、间隔棒(spacer)、防震锤(damper)和均衡环(ring)共4类目标,数据集中训练集图片30000张,测试集图片10000张。通过图像标注软件生成记录目标位置坐标信息的xml标签文件[20]。
3.2 实验结果及分析实验采用的预训练基础模型为VGG16,初始学习率为0.0001;迭代训练6万次和8万次后学习率分别降低为0.00001和0.000001,使模型更容易收敛;共训练90 000次,动量为0.9,权重衰减为0.000 5,输入模型中的图像大小为300像素×300像素;当迭代到40 000次时训练趋于稳定。其训练损失统计如图5所示。
Download:
|
|
本文针对Faster RCNN、SSD和本文提出的amSSD模型在测试集上进行对比实验,评价指标为准确率(average precision, AP)、平均准确率(mean average precision, mAP)和识别速率(frames per second, FPS),得到如表1、2所示的识别结果,部分实验结果对比如图6、7所示。
Download:
|
|
Download:
|
|
从表1中可以看出,针对4种电力部件绝缘子、间隔棒、防震锤和均衡环,SSD算法的识别准确率分别是88.6%、82.9%、79.1%和82.3%;Faster RCNN算法的识别准确率分别是90.1%、85.6%、75.2%和80.4%;而本文提出的模型识别准确率分别是92.2%、90.6%、87.6%和87.5。从表2中可以看出,Faster RCNN的mAP为82.5%,速度为16 FPS;SSD算法的mAP为83.4%,速度为47 FPS;本文算法的mAP为89.6%,速度为38 FPS。其中,Faster RCNN的识别准确率和识别速度最低,本文算法识别准确率最高,SSD算法的识别速度最快。本文算法比SSD算法平均识别准确率提高了6.2%,针对较小的电力部件如间隔棒和防震锤,识别率分别有7.7%和8.5%的较大提升,提升效果明显。同时本文算法的识别速度为38 FPS,相对原始SSD识别速度下降了9 FPS。这是由于amSSD模型增加了膨胀卷积层、SENet结构和反卷积层,导致计算量增加,但算法仍能满足输电线路巡检实时性的要求。因此,综合评价部件识别率和巡检系统的实时性能,本文提出的算法性能是最优的。
4 结论本文以SSD网络为基础,通过在特征提取阶段引入通道注意力机制和多尺度特征融合,提出了一种改进的amSSD输电线路电力部件识别网络。实验表明本文提出的网络在输电线路电力部件数据集上的识别性能与原始SSD和Faster RCNN相比更加优异,识别速度相较于SSD有一定下降,但仍能满足实时检测要求。
1)与原始SSD和Faster RCNN相比,本文提出的改进模型针对较小尺寸的电力部件具有更低的漏检率和更高的准确率。同时,尺寸较大的电力部件保持着良好的识别和定位效果。能够为后续输电线路故障检测打下坚实的基础。
2)本文的进一步工作将对损失函数进行改进,增大难分负样本的权重占比。在此基础上,继续进行电力部件故障检测算法的研究。
[1] | 张秋雁, 杨忠, 姜遇红, 等. 一种基于CNN的航拍输电线路图像分类方法[J]. 应用科技, 2019, 46(6): 41-45. (0) |
[2] | 李军华, 权小霞, 汪宇玲. 多特征融合的瓷砖表面缺陷检测算法研究[J]. 计算机工程与应用, 2020, 56(15): 191-198. (0) |
[3] | 付波, 徐超, 赵熙临, 等. 基于最值平均的人脸识别LBP算法[J]. 计算机应用与软件, 2019, 36(9): 209-213. DOI:10.3969/j.issn.1000-386x.2019.09.037 (0) |
[4] | 刘兴长, 孟昱煜. 基于HOG特征提取和模糊支持向量机的西夏文字识别[J]. 西北师范大学学报(自然科学版), 2019, 55(5): 39-43. (0) |
[5] | 林刚, 王波, 彭辉, 等. 基于改进Faster-RCNN的输电线巡检图像多目标检测及定位[J]. 电力自动化设备, 2019, 39(5): 213-218. (0) |
[6] | REN S, HE K, GIRSHICK R, et al. Faster R-CNN: towards real-time object detection with region proposal networks[J]. IEEE transactions on pattern analysis and machine intelligence, 2017, 39(6): 1137-1149. DOI:10.1109/TPAMI.2016.2577031 (0) |
[7] | LIU Wei, ANGUELOV D, ERHAN D, et al. SSD: Single shot multibox detector[C]//Proc of the 14th European Conference on Computer Vision. Amsterdam, the Netherlands, 2016: 21-37. (0) |
[8] | REDMON J, DIVVALA S, GIRSHICK R, et al. You only look once: unified, real-time object detection[C]// Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition.Las Vega, USA, 2016:779-788. (0) |
[9] | FU C Y, LIU W, RANGA A, et al. DSSD : deconvolutional single shot detector[C]//Computer Vision and Pattern Recognition. Hawaii, USA, 2017:2881-2890. (0) |
[10] | 刘涛, 汪西莉. 采用卷积核金字塔和空洞卷积的单阶段目标检测[J]. 中国图象图形学报, 2020, 25(1): 102-112. DOI:10.11834/jig.190166 (0) |
[11] | 刘学平, 李玙乾, 刘励, 等. 嵌入SENet结构的改进YOLOV3目标识别算法[J]. 计算机工程, 2019, 45(11): 243-248. (0) |
[12] | 彭晏飞, 梅金业, 王恺欣, 等. 基于区域注意力机制的遥感图像检索[J]. 激光与光电子学进展, 2020, 57(10): 180-188. (0) |
[13] | HU J, SHEN L, SUN G. Squeeze-and-excitation networks[C]// Conference on Computer Vision and Pattern Recognition. Salt Lake City, USA, 2018: 7132-7141. (0) |
[14] | TAO X, ZHANG D, WANG Z, et al. Detection of power line insulator defects using aerial images analyzed with convolutional neural networks[J]. IEEE transactions on systems, man, and cybernetics: systems, 2018: 1-13. (0) |
[15] | 马健, 史文旭, 鲍胜利. 基于特征融合SSD的遥感图像舰船目标检测[J]. 计算机应用, 2019, 39(S2): 253-256. (0) |
[16] | 陈庆, 闫斌, 叶润, 等. 航拍绝缘子卷积神经网络检测及自爆识别研究[J]. 电子测量与仪器学报, 2017, 31(6): 942-953. (0) |
[17] | 徐诚极, 王晓峰, 杨亚东. Attention-YOLO:引入注意力机制的YOLO检测算法[J]. 计算机工程与应用, 2019, 55(6): 13-23, 125. (0) |
[18] | 赵文清, 程幸福, 赵振兵, 等. 注意力机制和Faster RCNN相结合的绝缘子识别[J]. 智能系统学报, 2020, 15(1): 92-98. (0) |
[19] | RUSSAKOVSKY O, DENG J, SU H, et al. ImageNet large scale visual recognition challenge[J]. International journal of computer vision, 2014, 115(3): 211-252. (0) |
[20] | 颜宏文, 陈金鑫. 基于改进YOLOv3的绝缘子串定位与状态识别方法[J]. 高电压技术, 2020, 46(2): 423-432. (0) |