来源:期刊VIP网 时间:
作者:秦伦明;王朝举;边后琴;崔昊杨;王悉;
单位:上海电力大学电子信息工程学院;北京交通大学电子信息工程学院;
摘要:为解决目前语义分割算法在电力线分割领域存在预测速度缓慢和分割精度不高的双重问题,提出一种电力线分割网络STDC-DeepLabv3+。首先,为提升网络预测速度,在编码器部分采用Swin Transformer V2设计轻量化的主干特征提取网络;其次,为提高分割精度,针对电力线的细长结构以及贯穿整幅图片的特点,提出动态蛇形空间金字塔池化(DSASPP)模块,同时,在解码器部分设计多尺度特征融合模块,使网络更好地利用不同层次的语义信息提取电力线特征,减少网络对电力线的漏分割现象;最后,引入坐标注意力(CA)机制减少背景干扰,进一步提升分割的准确率。实验结果表明,改进后的算法平均交并比(MIoU)和平均像素精度(MPA)分别达到了84.18%和92.85%,与现有分割算法相比,分割精度和预测速度均有所提升,预测速度与DeepLabv3+相比提升了93.92%。
关键词:电力线分割;;Transformer;;DeepLabv3+;;多尺度特征融合;;编码器;;解码器;;坐标注意力机制
基金资助:国家自然科学基金面上项目(62073024)