基于改进U-Net的轻量级输电线分割算法
DOI:
CSTR:
作者:
作者单位:

1.东北大学机械工程与自动化学院沈阳110819;2.哈尔滨工业大学仪器科学与工程学院哈尔滨150001

作者简介:

通讯作者:

中图分类号:

TP391.4;TN98

基金项目:

国家电网有限公司总部科技项目(52170218000G)资助


Lightweight transmission line conductor segmentation algorithm with improved U-Net
Author:
Affiliation:

1.School of Mechanical Engineering & Automation, Northeastern University, Shenyang 110819, China; 2.School of Instrumentation Science and Engineering, Harbin Institute of Technology, Harbin 150001, China

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    为了提高输电线路的巡检效率,保证输电线的分割精度和速度,本文提出基于改进U-Net的轻量级网络GU-Net。首先,以U-Net网络为基础,在编码器部分引入轻量化主干提取网络GhostNet;然后采用双线性插值方法完成上采样,并利用深度可分离卷积代替部分普通卷积;最后在训练过程引入多损失函数以解决输电线和背景像素占比不平衡问题,并采用迁移学习策略训练模型。在E-Wire输电线数据集上测试,GU-Net网络的MIoU和F1-score分别为80.04%和87.77%,与现有的轻量化输电线语义分割网络Wire-Detection相比分别提升了4.26%和2.96%,且分割速度几乎没有损失,参数量约是它的20%。实验结果表明,本文提出的算法能够实现快速高效、轻量化地分割出复杂图像中的输电线。

    Abstract:

    To improve the inspection efficiency of transmission lines and ensure the segmentation accuracy and speed of transmission lines, this paper proposes GU-Net, a lightweight network based on improved U-Net. Firstly, based on the U-Net network, the lightweight trunk extraction network Ghost-Net is introduced in the encoder part; then a bilinear interpolation method to complete the up-sampling and use the depth-separable convolution to replace part of the ordinary convolution; finally, introduce multiple loss functions in the training process to solve the imbalance between the transmission line and the background pixel occupancy, and train the model with a migration learning strategy. Tested on the E-Wire transmission line dataset, the MIoU and F1-score of the GU-Net network are 80.04% and 87.77%, respectively, which are 4.26% and 2.96% better than Wire-Detection, an existing semantic segmentation network for lightweight transmission lines, with almost no loss in the segmentation speed, and the number of references is about 20% of it. The experimental results show that the algorithm proposed in this paper can achieve fast, efficient and lightweight segmentation of transmission lines in complex images.

    参考文献
    相似文献
    引证文献
引用本文

胡冠华,张永雷,申立群.基于改进U-Net的轻量级输电线分割算法[J].电子测量与仪器学报,2024,38(1):211-218

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2024-04-03
  • 出版日期:
文章二维码