郑州大学学报(理学版)  2019, Vol. 51 Issue (4): 16-22  DOI: 10.13705/j.issn.1671-6841.2018299

引用本文  

高金峰, 吕易航. 航拍图像中绝缘子串的识别与分割方法研究[J]. 郑州大学学报(理学版), 2019, 51(4): 16-22.
GAO Jinfeng, LÜ Yihang. Research on Recognition and Segmentation of Insulator Strings in Aerial Images[J]. Journal of Zhengzhou University(Natural Science Edition), 2019, 51(4): 16-22.

基金项目

国家自然科学基金项目(61876169)

通信作者

吕易航(1993—),男,河南郑州人,硕士研究生,主要从事电工理论与新技术研究,E-mail:xmslvyihang@163.com

作者简介

高金峰(1963—),男,河南郑州人,教授,主要从事电工理论与新技术、信号检测与处理、电子技术应用等研究,E-mail:jfgao@zzu.edu.cn

文章历史

收稿日期:2018-11-07
航拍图像中绝缘子串的识别与分割方法研究
高金峰, 吕易航    
郑州大学 电气工程学院 河南 郑州 450001
摘要:为实现航拍图像中绝缘子串的识别与分割,以Faster R-CNN和全卷积神经网络(fully convolutional networks, FCN)为框架构建了深度学习算法,并以算法为核心搭建了绝缘子串图像检测平台.算法首先使用Faster R-CNN模型,结合Resnet-101深度残差卷积神经网络,对所有感兴趣区域进行分类,再对Bounding Box回归和坐标进行修正,用于绝缘子串的识别,接着以FCN网络为基础,通过微调使原网络适应新的绝缘子串数据集,实现了在复杂背景下绝缘子串的语义分割.实验结果表明,文中给出的方法与已有方法相比,不仅能够在不同光照条件、不同拍摄角度以及复杂背景干扰下实现绝缘子串的识别与分割,且处理时间短、精度高和鲁棒性强.
关键词绝缘子串    Faster R-CNN    FCN    深度学习    绝缘子串识别    语义分割    
Research on Recognition and Segmentation of Insulator Strings in Aerial Images
GAO Jinfeng, LÜ Yihang    
School of Electrical Engineering, Zhengzhou University, Zhengzhou 450001, China
Abstract: In order to realize the identification and segmentation of insulator strings in aerial images, a deep learning algorithm was proposed by using Faster R-CNN and FCN as the core framework to form an image detection platform. With the combination of the Faster R-CNN and the Resnet-101, the regions of interest were classified, and then Bounding Box regression and coordinate correction were made in order to achieve insulator strings detection. The FCN was fine-tunned on, to adapt to the new-insulator dataset, which could realize the segmentation of the insulator strings in complex background. Compared with the existing methods, the method presented not only could realize the identification and segmentation of insulator strings in different illumination conditions, different shooting angles and complex background interference, but also could have short processing time, high precision and strong robustness.
Key words: insulator strings    Faster R-CNN    FCN    deep learning    insulator strings detection    semantic segmentation    
0 引言

对输电线路进行定期检查,随时掌握输电线路的运行情况,是电网运检人员的日常工作之一.使用无人机进行线路巡检具有方式灵活、成本低等优势[1].无人机搭载的摄像头可提供大量的高清图片,图片中的绝缘子串是线路巡检中关注的重要对象.如何在多种拍摄角度、光照条件以及背景噪声等影响下进行绝缘子串的识别和分割具有重要意义.

国内外学者关于绝缘子串的识别与分割方法已有一些研究,有些在实际生产中得到了应用.文献[2]使用HOG特征加SVM分类器.文献[3]通过搭建CNN并选用SVM分类器,文献[4]使用Alex Net、VGG16等框架分别训练分类器和检测器并将其级联,分别实现了绝缘子串的识别和定位.文献[5]提出了一种基于深度学习U-net网络的航拍绝缘子串分割方法,进行像素提取和定位,提高了定位精度.文献[2-4]均采用特征提取器加分类器的方法,识别速度较慢,精度较低.文献[5]中的深度学习算法U-net目前主要应用于样本较小的医学影像的处理,在其他领域的应用较少.

国内外学者也开展了通用图像的分割方法的研究.文献[6]提出了一种基于三维OTSU的图像分割方法,该方法计算复杂度较高,分割效率低.文献[7]提出了一种基于Prewitt算子的边缘检测算法,可以有效抑制噪声且较好地解决了边缘粗糙等问题.文献[8]提出了小波模极大值的图像分割算法,最大程度地减少了噪音,该方法在抑制噪音的同时也削弱了目标的信息强度,不利于图像的分割.

本文以深度学习算法为依托,采用速度快、效率高的Faster R-CNN[9]和FCN[10]模型,构造了一种识别与分割方法,搭建了图像处理平台.实际应用表明,该方法能够在不同光照条件、不同拍摄角度以及复杂背景噪声干扰下的航拍图像中实现绝缘子串的识别与分割,且处理时间短、精度高、鲁棒性强.

1 绝缘子串识别与分割平台

深度学习算法是目前机器学习算法研究中最受关注且取得较大成功的算法之一,其核心就是使用多层或深度的网络,建立类似于人脑学习机制的复杂神经网络.搭建的神经网络具有拟合任意复杂函数的能力,构建复杂的非线性映射,可以将观察数据进行分层特征表示,将低级特征抽象化后形成高级特征表示.深度学习在图像处理和计算机视觉应用中都取得了令人满意的进展.

在图像分类领域,AlexNet[11]、VGG(http://arxiv.org/abs/1409.1556)、GooLeNet[12]以及ResNet[13]网络在ImageNet数据集上实现了1 000多类图像的分类,Resnet网络将Top-5错误率降到了3.57%.

在图像检测领域,R-CNN[14]、Fast R-CNN[15]、Faster R-CNN网络在VOC-2007数据集上实现了20类物体的识别,mAP达到了66.9%;另外在FDDB人脸数据集中,可以识别出5 171张人脸.

在图像分割领域,FCN网络在VOC-2012数据集上实现了生活中20类物体像素级别的语义分割,FCN-8s分割的像素精度达到了90.3%.

本文为方便得到绝缘子串识别与分割结果,将构建的深度学习算法与数据预处理等步骤集成在一起,形成绝缘子串检测平台.该平台只需将所要检测的航拍图片导入,就能自动处理,得出绝缘子串识别与分割结果.该平台依托后台图形计算服务器,其配置是Window10系统、两个Intel酷睿8代CPU、主频3.2 GHz、6核心、12线程、64 G内存、Nvidia Geforce GTX Titan Xp图形处理器,12 G显存.深度学习框架为Tensorflow和Caffe的python接口.

本文所使用的数据集包括两部分:一部分来自Github的Insulator Data Set-Chinese Power Line Insulator Dataset(CPLID)(https://github.com/InsulatorData/InsulatorDataSet.);另一部分来自无人机航拍视频的逐帧图片.数据集共2 000张图片,这些图片全部由无人机在不同背景下,从不同角度拍摄的.绝缘子串航拍图像中包含复杂的背景噪声,如天空、森林、田地、房屋、山脉等.另外,由于拍摄角度不同,在图像中绝缘子串还会出现遮挡甚至重叠的现象.不同背景下航拍图像中的绝缘子串如图 1所示.

图 1 不同背景下航拍图像中的绝缘子串 Fig. 1 The insulator strings image with different aerial backgrounds
2 图像识别算法及其实现 2.1 Faster R-CNN算法

Faster R-CNN是在R-CNN、SPP-NET、Fast R-CNN网络的基础上改进得来的.该算法主要解决两个问题:一是提出区域建议网络RPN,快速生成候选区域;二是通过交替训练,使RPN和Fast R-CNN网络共享参数.Faster R-CNN关于一个图像的损失函数定义为

$L\left(\left\{p_{i}\right\}, \left\{u_{i}\right\}\right)=\frac{1}{N_{\mathrm{cls}}} \sum\limits_{i} L_{\mathrm{cls}}\left(p_{i}, p_{i}^{*}\right)+\lambda \frac{1}{N_{\mathrm{reg}}} p_{i}^{*} L_{\mathrm{reg}}\left(t_{i}, t_{i}^{*}\right), $

式中:pi为anchor预测目标的概率;GT标签$ p_{i}^{*}=\left\{\begin{array}{ll}{0} & {\text { negative label }} \\ {1} & {\text { positive label }}\end{array}\right. $ti={tx, ty, tw, th}表示预测的Bounding Box 4个顶点的参数化坐标;Lcls(pi, pi*)为两个类别(目标,背景)的对数损失,所用公式为Lcls(pi, pi*)=-log[pi*pi+(1-pi*)(1-pi)];Lcls(ti, ti*)为回归损失,所用公式为Lcls(ti, ti*)=R(titi*).

RPN是使用卷积神经网络直接产生区域建议,本质上就是用滑动窗口在最后的卷积层上滑动一遍.RPN可以针对生成检测建议框的任务端到端地训练,能够同时检测出目标的边界和分数.

有些RPN推荐框和其他推荐框大量重叠,为了减少冗余,可基于推荐区域的cls得分,采取非极大值抑制,设定恰当的阈值,留下2 000个anchor,取前N个给Faster R-CNN,对分类数进行筛选,并取分数大于某个分数的目标结果.Faster R-CNN的训练流程如图 2所示.

图 2 Faster R-CNN训练流程图 Fig. 2 Training process of Faster R-CNN
2.2 Faster R-CNN检测的实现 2.2.1 数据的预处理

将2 000张航拍图统一成640×480大小的样本,并打乱顺序,随机分布.然后利用Keras框架进行数据增强,如翻转、随机修剪、色彩抖动、平移变换、尺度变换、对比度变换、噪声扰动等,新增1 000张样本图片,合计3 000张图片.

将3 000张样本图片用开源的labelImg(https://github.com/tzutalin/labelImg)进行标注,生成groundtruth.将总样本的80%作为训练集,20%作为测试集.

2.2.2 Faster R-CNN的训练与调参

具体操作如下.

1) 训练RPN网络.新增的2层参数用均值为0、标准差为0.01的高斯分布来进行初始化,其余层用ImageNet预训练的模型初始化,并端到端微调,用于区域建议任务.

2) 利用1)中的RPN生成的建议框,由Fast R-CNN训练一个单独的检测网络,同样由ImageNet预训练模型初始化,此时,两个网络并没有共享卷积层.

3) 用检测网络初始化RPN,固定共享卷积层,微调RPN独有的层.

4) 固定共享的卷积层,微调Fast R-CNN的Fc层,两个网络共享相同的卷积层.

训练参数设置如下:最大迭代次数为200万次;初始学习率为0.001;学习率衰减策略为‘step’;每迭代1万次学习率进行一次衰减;一个mini_batch包含2张图片;Minibatch=128;代用dropout避免过拟合,dropout=0.2,采用随机梯度下降法进行权重更新,交替训练的各个阶段迭代1万次;RPN正样本的阈值为0.7,负样本的阈值为0.3;NMS阈值为0.7;每训练1 000次进行一次测试,test_size=2.

图 3所示,迭代200万(2 000×1 000)轮之后,选取在同一杆塔上,两种不同角度拍摄的绝缘子串图片,识别效果良好.

图 3 绝缘子串识别效果图 Fig. 3 Pictures of insulator strings detection
3 图像分割算法及其实现 3.1 FCN算法

语义分割是从像素水平上理解、识别图片的内容.输入的是一张图片,输出的是同尺寸的分割标记(像素水平上),而且每个像素都会被识别为一个类别.

利用全卷积网络对图像进行像素级别的分类,从而解决了语义级别的图像分割问题.与经典的CNN不同,FCN可以接受任意尺寸的图像,采用反卷积层对最后一个卷积层的feature map进行上采样,使它恢复到与输入图像相同的尺寸,从而可以对每个像素都产生一个预测,同时保留了原始图像中的空间信息,最后在上采样的特征图上进行逐个像素分类.主要有3层结构:

一是卷积层,将所有CNN的全连接层转换成卷积层,它能适应任意尺寸输入,并输出低分辨率分割图片.CNN网络主要有Alexnet、VGG16和GoogleNet,输出的图片经过5个卷积层,经过每个卷积层后的分辨率降低为原来的1/2,经过5个卷积层后分辨率就降低至原来的1/32.

二是反卷积,也叫转置卷积.将低分辨率图片进行上采样,输出同分辨率分割图片.反卷积就是卷积的逆操作,也是一对多的操作,其中卷积核的尺寸不变,对应于卷积操作的后向和前向传播,在优化上做了颠倒,而反卷积核是卷积核的转置,学习率为0.

三是跳层结构,可将同分辨率的图片边缘精化,直接用32倍反卷积得到的分割结果比较粗糙,所以可以使用前两个卷积层的输出做融合,pool3和pool4后会增加一个1×1卷积层做预测,较浅网络的结果精细,较深网络的结果鲁棒.

3.2 FCN语义分割的实现 3.2.1 数据预处理

为减轻标签制作时的困难,在数据增强后的3 000张图片中选取1 000张没有遮挡或重叠的单排或双排绝缘子串作为候选.制作Groundtruth时,利用开源的LabelMe[16]进行标注,将生成的json文件转化成png文件索引图(训练时作为输入).

1 000张图片的80%作为训练集,20%作为测试集.

3.2.2 参数设置

最大迭代次数为25万次;初始学习率为0.000 1;学习率衰减策略为‘step’;每迭代1万次学习率进行一次衰减;一个mini_batch包含2张图片;用dropout避免过拟合,dropout=0.1;每训练1.25万次进行一次测试,test_size=2;每经过1万次迭代产生一次snapshot,共产生10个snapshot.

3.2.3 loss和accuracy曲线可视化

经过25万轮迭代之后,由caffe可视化训练损失、测试准确率与迭代次数的变化曲线如图 4所示.

图 4 训练损失和测试准确率的变化曲线 Fig. 4 Discharging of train loss and test accuracy

图 4可知,经过25万次迭代之后训练损失最终收敛到0.4附近;在每迭代1.25万次时自动测试一次,在25万轮过后,测试准确率达到了约97%.

随机选取2张玻璃和瓷质绝缘子串的航拍图片,分割结果可视化如图 5所示.

图 5 绝缘子串分割效果图 Fig. 5 Pictures of insulator strings segmentation
4 实验结果 4.1 航拍图中绝缘子串的识别结果

在绝缘子串检测平台上,为了验证本文所用算法的识别效果,选取多目标的航拍图来进行验证,将本文所用的Faster R-CNN算法与文献[2]和文献[3]算法作对比,其识别结果如表 1所示.图像(a)是无人机顺光时对杆塔的拍摄,对应了浅色的绝缘子串和深色的背景;图像(b)是无人机在逆光时对杆塔的拍摄,对应了深色的绝缘子串和浅色的背景.

表 1 绝缘子串识别结果对比 Tab. 1 Comparison of the results of insulator strings detection

表 1可知,在深色背景下,两种算法生成的矩形框杂乱不堪,既有误识别现象,又有重复识别现象;在浅色背景下,重复识别现象略有减少,但是误识别现象还时有发生.而基于深度学习的Faster R-CNN算法既没有误识别现象,也没有重复识别现象.

4.2 航拍图中绝缘子串的分割结果

为验证本文所用的语义分割算法的有效性,随机选取3张绝缘子串航拍图(包含浅色背景和深色背景),分别用文献[6]提出的OTSU算法、文献[7]提出的prewitt算子边缘检测算法、文献[8]提出的小波模极大值算法和本文使用的FCN算法对3张航拍图分割提取,分割结果如表 2所示.

表 2 绝缘子串分割结果对比 Tab. 2 Comparison of the results of insulator strings segmentation

首先,就分割后划分背景和目标(绝缘子串)的色调分析,由表 2可知,OTSU算法会自动选取前景(目标)和背景的分割阈值,OTSU算法将图像(c)二值化之后,背景和绝缘子串分别为白色和黑色,将图像(d)和(e)二值化之后,背景和绝缘子串分别为黑色和白色;小波模极大值算法会分尺度将图像分解为各个频率下的分量,不同尺度可以检测不同频率的图像局部特征,所以当背景和目标色调强度差别不大时,分割出来的效果成浅色调,掩盖了图像的主体信息;prewitt算子边缘检测算法能较好地区分背景和目标(绝缘子)的色调.其次,就抗噪能力分析,OTSU算法、小波模极大值算法和prewitt算子边缘检测算法的分割结果都不能良好地除去背景噪声.

本文采用的FCN语义分割算法,不但能够清晰地区分出背景和目标的色调,而且能精确地分割出复杂背景中的绝缘子串,抗噪能力强,有着更广泛的使用性和鲁棒性.

4.3 Faster R-CNN算法和FCN算法的性能测试

在上述实验平台下,绝缘子串识别处理时间对比见表 3,绝缘子串分割处理时间对比见表 4,绝缘子串识别和分割结果见表 5.

表 3 绝缘子串识别处理时间对比 Tab. 3 Comparison of the running time of the insulator strings detection 

表 4 绝缘子串分割处理时间对比 Tab. 4 Comparison of the running time of the insulator strings segmentation 

表 5 绝缘子串的识别和分割结果 Tab. 5 The results of the insulator strings detection and segmentation

表 3可知,本文Faster R-CNN算法对图像(a)和图像(b)的处理时间小于文献[2]的Hog特征+SVM算法和文献[15]的CNN+SVM算法处理时间.

表 4可知,本文FCN算法对于图像(c)、图像(d)和图像(e)的处理时间要小于文献[6]的OTSU算法、文献[7]的prewitt算子边缘提取算法和文献[8]的小波模极大值算法.对于图像(e)(像素为1 500×1 000)这种高分辨率的航拍图来说,FCN算法的处理时间能维持在1 s左右,可见其性能的优势.

表 5可知,本文Faster R-CNN算法能够识别600张绝缘子串航拍图中的581张,其识别率达到了96.83%;FCN算法能够分割出200张绝缘子串航拍图中的181张,准确率达到了90.50%.

综上所述,本文基于深度学习的两种算法对于绝缘子航拍图的处理性能优于其他传统算法.

5 结论

本文以航拍图像中绝缘子串的识别与分割为目标,分别利用Faster R-CNN与FCN构建了绝缘子串识别与分割算法,使用Tensorflow和Caffe的python接口搭建了基于深度学习的绝缘子串检测平台,对来自github和无人机航拍视频中截取的图片进行了训练和测试,以本文算法为基础搭建的绝缘子串检测平台的识别率达到了96.83%,分割率达到了90.50%,提高了不同光照条件、拍摄角度和复杂背景噪声的抗干扰水平,且处理速度快.

参考文献
[1]
TOTH J, GILPIN-JACKSON A. Smart view for a smart grid: unmanned aerial vehicles for transmission lines[C]//1st International Conference on Applied Robotics for the Power Industry. Montreal, 2010, 1-6. (0)
[2]
文学志, 方巍, 郑钰辉. 一种基于类Haar特征和改进AdaBoost分类器的车辆识别算法[J]. 电子学报, 2011, 39(5): 1121-1126. (0)
[3]
陈庆, 闫斌, 叶润, 等. 航拍绝缘子卷积神经网络检测及自爆识别研究[J]. 电子测量与仪器学报, 2017, 31(6): 942-953. (0)
[4]
侯春萍, 章衡光, 张巍, 等. 输电线路绝缘子自爆缺陷识别方法[J]. 电力系统及其自动化学报, 2019, 31(6): 1-6. (0)
[5]
陈景文, 周鑫, 张蓉, 等. 基于U-net网络的航拍绝缘子检测[J]. 陕西科技大学学报, 2018, 36(4): 153-157. DOI:10.3969/j.issn.1000-5811.2018.04.028 (0)
[6]
景晓军, 李剑峰, 刘郁林. 一种基于三维最大类间方差的图像分割算法[J]. 电子学报, 2003, 31(9): 1281-1285. DOI:10.3321/j.issn:0372-2112.2003.09.001 (0)
[7]
康牧, 许庆功. 基于prewitt理论的自适应边缘检测算法[J]. 计算机应用研究, 2009, 26(6): 2383-2386. DOI:10.3969/j.issn.1001-3695.2009.06.116 (0)
[8]
曹文明, 王耀南, 印峰, 等. 高压输电线路除冰机器人障碍物识别方法研究[J]. 仪器仪表学报, 2011, 32(9): 2049-2056. (0)
[9]
REN S Q, HE K M, 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)
[10]
SHELHAMER E, LONG J, DARRELL T. Fully convolutional networks for semantic segmentation[J]. IEEE transactions on pattern analysis and machine intelligence, 2017, 39(4): 640-651. DOI:10.1109/TPAMI.2016.2572683 (0)
[11]
KRIZHEVSKY A, SUTSKEVER I, HINTON G E. ImageNet classification with deep convolutional neural networks[J]. Communications of the ACM, 2017, 60(6): 84-90. DOI:10.1145/3065386 (0)
[12]
SZEGEDY C, LIU W, JIA Y Q, et al. Going deeper with convolutions[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition. Piscataway, 2015: 1-9. (0)
[13]
HE K M, ZHANG X Y, REN S Q, et al. Deep residual learning for image recognition[C]//IEEE Conference on Computer Vision and Pattern Recognition. Las Vegas, 2016: 770-778. (0)
[14]
GIRSHICK R, DONAHUE J, DARRELL T, et al. Rich feature hierarchies for accurate object detection and semantic segmentation[C]//Proceedings of IEEE Conference on Computer Vision and Pattern Recognition. Berkeley, 2014: 580-587. (0)
[15]
GIRSHICK R. Fast R-CNN[C]//IEEE International Conference on Computer Vision. Santiago, 2015: 1440-1448. (0)
[16]
RUSSELL B C, TORRALBA A, MURPHY K P, et al. LabelMe: a database and web-based tool for image annotation[J]. International journal of computer vision, 2008, 77(1/2/3): 157-173. (0)