基于改进 U-net 网络的液压管路分割方法
作者:
作者单位:

作者简介:

通讯作者:

中图分类号:

TP391. 4 ;TH137. 8 + 6

基金项目:

陕西省液压技术重点实验室开放基金(YYJS2022KF08)、陕西省工业科技攻关项目 (2015GY068)资助


Hydraulic pipeline segmentation method based on improved U-net network
Author:
Affiliation:

Fund Project:

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

    针对液压管路背景多变、管道弯折、管道重叠排布等复杂现象且现有图像分割方法对管路分割精度不高等问题,提出一 种以 U-net 网络为基础,结合 Mobilenetv3 网络、 SE 注意力机制模块、自校正卷积模块的液压管路分割方法。 该方法以 Mobilenetv3-large 模型作为骨干网络,结合 LR-ASPP 网络处理特征图;在解码过程中,融入 SE 注意力模块和 SC 自校正模块,提 升了特征提取能力;最后采用 Dice 函数和 BCE 函数的组合来作为网络的损失函数,有效地提升了网络的收敛能力。 实验结果 表明本文提出的方法在交并比、像素精度指标上的均值分别达到 90. 8%、95. 2%,且模型体积为 16. 9 M,推理每张图像所耗时间 20 ms,可应用于需实时部署的场景,为液压管路渗漏的准确识别提供了基础。

    Abstract:

    Aiming at the complex phenomena such as variable background, bending and overlapping arrangement of hydraulic pipeline and the low accuracy of pipeline segmentation by existing image segmentation methods, a hydraulic pipeline segmentation method based on U-net network, combined with Mobilenetv3 network, squeeze-and-excitation networks module and self-calibration convolutional module is proposed. The method selected Mobilenetv3-large model as the backbone network, and the feature maps are processed with Lraspp network. In the decoding process, the squeeze-and-excitation networks module and self-correction module are integrated to improve the feature extraction ability. Finally, the combination of Dice function and BCE function is used as the loss function of the network, which effectively improves the convergence ability of the network. Experimental results show that the mean values of the proposed method in the intersection over union and pixel accuracy are 90. 8% and 95. 2%, respectively. The model size is 16. 9 M, and the reasoning time for each image is 20 ms, which can be applied to the scene requiring real-time deployment. It provides a basis for the accurate identification of hydraulic pipeline leakage.

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

张 平,佟昆宏,王学珍.基于改进 U-net 网络的液压管路分割方法[J].电子测量与仪器学报,2023,37(1):123-129

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