一种基于UNet网络的复杂背景下输电线识别方法

摘要

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