[发明专利]一种基于UNet网络的复杂背景下输电线识别方法

说明书

一种基于UNet网络的复杂背景下输电线识别方法，包括以下步骤：步骤1，获取不同视角含有输电线图像；步骤2，对含有输电线图像进行预处理，即采用数据增强方式扩充含有输电线图像，建立含有输电线图像数据集，给每张含有输电线图像中的输电线记标签，获得输电线像素为1，背景像素为0的标签图；步骤3，构建UNet网络；步骤4，把含有输电线图像数据集中的图像及对应的标签图输入UNet网络进行训练，获得最优的权重参数及对应的最优输电线识别模型；步骤5，将最优输电线识别模型应用于需要识别的输电线图像，识别提取出输电线。本发明抗干扰能力强，稳定性高，能够在各种复杂背景含有输电线图像中，精准的识别出输电线，识别效果好。

技术领域：

本发明涉及计算机视觉领域，具体涉及一种基于UNet网络的复杂背景下输电线识别方法。

背景技术：

输电线因长期暴露在野外环境中易发生损坏，如果不及早发现并消除隐患，将导致国民经济重大损失。因此，为确保安全，电力公司需对输电线路进行定期巡检。目前无人机技术的高速发展为输电线路巡检带来了新的前景。无人机体积小，成本低，操控方便，可以翻山越岭在艰苦恶劣环境下飞行，适用于在不同位置和不同角度拍摄图像和视频，具有优良的机动性和灵活性。

如何从复杂多样的航拍背景图像中完整、准确地提取输电线，是无人机电力巡检必须解决的关键问题。目前，许多学者针对航拍图像输电线提取方法进行了大量研究。其中，利用抗噪性能较好的Ratio算子进行输电线的边缘检测，由于Ratio算子利用中间区域和两边相邻区域灰度均值的比值来进行边缘检测，必须严格限定图像中输电线的方向，并以此设定Ratio算子的检测方向，因约束条件苛刻，实际应用受到制约。利用多组形态学算法实现输电线边缘像素点的快速检测，但是鲁棒性较差，不同复杂背景下输电线提取效果差别较大。利用方向滤波进行自相关增强，抑制复杂背景的同时增强输电线目标，但是增强效果依赖于人工控制的迭代次数。利用Canny算子边缘检测，Hough变换提取直线，再经过直线段编组合并，此种方法在复杂背景下输电线识别效果差，容易提取出大量的噪声，影响后续直线段编组合并。

发明内容：

针对上述问题，本发明提供一种基于UNet网络的复杂背景下输电线识别方法，能够在各种复杂的背景下，不同视角含有输电线图像中，精确地识别出输电线。

为了实现本发明之目的，拟采用以下技术方案：一种基于UNet网络的复杂背景下输电线识别方法，包括以下步骤：

步骤1：获取不同视角含有输电线图像；

步骤2：对含有输电线图像进行预处理，即采用数据增强方式扩充含有输电线图像，建立含有输电线图像数据集，给每张含有输电线图像中的输电线记标签，获得输电线像素为1，背景像素为0的标签图；

步骤3：构建UNet网络；

步骤4：把含有输电线图像数据集中的图像及对应的标签图输入UNet网络进行训练，获得最优的权重参数及对应的最优输电线识别模型；

步骤5：将最优输电线识别模型应用于需要识别的输电线图像，识别提取出输电线。

本方案中所述步骤1中，在不同复杂背景下，航拍获取不同视角的含有输电线图像。

进一步地，所述步骤2中，数据增强方式采用随机翻转、裁剪、旋转扩充含有输电线图像，用标注软件将含有输电线图像中的输电线记标签，将含有输电线图像及对应的标签图按照8∶1∶1的比例随机分配生成训练集、验证集和测试集。

进一步地，所述步骤3中，UNet网络包括下采样编码、上采样解码和跳跃链接，下采样编码对输入图像进行多次卷积、归一化和最大池化操作，获得不同尺度的有效特征层，上采样解码对不同尺度的有效特征层进行两倍上采样恢复输入图像像素尺寸，跳跃链接中每一次下采样生成的有效特征层都会与对应的上采样进行特征融合，输出的特征层和输入图像的高宽相同。