关键词:
图像预处理
图像分割模型
半监督图像分割
摘要:
输电线路场景下的图像分割是一项重要的任务,可以作为野外输电线路的覆冰检测、树木或工程车的侵入检测、检测绘制电线上的电器组件和灾难监测等多项输电线路场景下安全隐患检测工作的前期任务。本文旨在利用数字图像处理技术和图像分割技术,研究适合工业场景应用部署的输电线路场景下的高精度图像分割模型。本研究能够对输电线路场景下的导线、杆塔以及地面实现精确的分割,为电力系统的隐患检测提供支撑,使电力系统安全、平稳地运行。图像分割任务数据标注成本高且本文使用的部分标注数据中存在标注错误的问题,因此本文采用半监督的图像分割方法,使用较少的标注数据达到精确的图像分割效果。本文主要从数据集预处理、图像分割模型以及半监督图像分割方法三个方面介绍所提出的基于半监督学习的输电线路场景图像分割方法。(1)在数据集预处理方面,本文使用的原始标注数据由已标注的三维点云数据转换得到,转换后的图像存在类别像素点离散的问题,不利于图像分割模型的训练并最终影响模型的检测效果。本文采用数字图像处理中形态学的方法,根据导线、杆塔以及地面类别的形状特点,分类别使用不同的预处理方法,使处理后的标注图像中,每一类别的像素点能够完全覆盖所属类别。同时,本文对输电线路场景图进行图像增强,使导线和杆塔特征更加明显,有效提高了导线和杆塔的分割效果。(2)在图像分割模型方面,为了进一步提升图像分割模型的效果,本文对图像分割模型中的特征提取网络进行改进,增加注意力机制使网络能够学习不同特征通道的重要性。针对导线类别像素宽度较小、图像分割模型难以分辨的问题,本文基于编码器-解码器结构的特点,结合自适应空间特征融合的原理,在图像分割模型的解码器中融合了更多浅层特征,在提升导线分割效果的同时使图像分割模型的整体分割精度也有所提高。(3)在半监督图像分割方法方面,本文以结合了自训练和一致性正则化两种方法的交叉伪监督框架(Cross Pseudo Supervision,CPS)为基础,分析了 CPS中自训练范式存在的问题:在模型训练前期,伪标签出现较多错误,误导图像分割模型的学习,并提出了基于随机数据增广的半监督图像分割框架,提升了图像分割模型训练的效果。为验证提出的基于半监督学习的输电线路场景图像分割方法的有效性,本文在输电线路场景真实数据集上进行大量实验并以平均交并比(Mean Intersection over Union,MIoU)作为衡量标准。在实验部分首先验证了本文提出的数据集预处理方法的有效性和必要性。通过对比实验,验证了本文提出的基于半监督学习的输电线路场景图像分割方法的优越性,与表现最好的基线方法相比,MIoU提升约4%。在使用1/2有标注数据的实验中导线、杆塔以及地面类别的IoU分别达到了 73.26%、77.58%以及88.12%。在对比试验后,本文设置一系列消融实验进一步证明了本文提出的各个改进方法的有效性。