一种基于生成对抗网络和改进U-Net的排水管道缺陷识别模型
作者:
作者单位:

1.四川省自然资源测绘地理信息有限责任公司,四川 成都 610212
2.西南石油大学 地球科学与技术学院,四川 成都 610500

作者简介:

刘勇(1977—), 男, 硕士, 正高级工程师, 从事测绘工程、 地下管网探测及供排水管网修复等工作。

通讯作者:

中图分类号:

TU992

基金项目:


A Defect Recognition Model for Drainage Pipes Based on Generative Adversarial Networks and Improved U-Net
Author:
Affiliation:

1.Sichuan Natural Resources Mapping and Geographic Information Co., Ltd., Chengdu 610212, China
2.School of Geosciences and Technology, Southwest Petroleum University, Chengdu 610500, China

Fund Project:

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

    通过集成生成对抗网络图像生成及增强技术,利用卷积注意力模块CBAM (Convolutional Block Attention Module)改进U-Net网络,提出了一种下排水管道缺陷高精度识别模型,以准确自动对排水管道影像缺陷进行分割识别。该模型取得了满意的识别结果,在验证集中平均像素精度mPA (Mean Pixel Accuracy)、平均交联度mIoU (Mean Intersection over Union)分别为91.40%、85.02%。结果表明,利用生成对抗网络对采集数据中较少的缺陷类别图像生成及增强,能够大大提高少样本缺陷识别的准确度,数据增强后对PSPNet、U-Net以及不同主干网络的DeepLabv3+网络得到大幅度提升,mPA、mIoU平均增长20%~30%。加入CBAM注意力模块能够提升U-Net的分割能力,使模型具有更好的像素级分割能力。

    Abstract:

    By integrating GAN (Generative Adversarial Network) for image generation and enhancement, and improving the U-Net network with the Convolutional Block Attention Module (CBAM), a high-precision recognition model for drainage pipeline defects is proposed to accurately and automatically segment and identify the image defects of drainage pipes. The model achieves satisfactory recognition results. The mPA (Mean Pixel Accuracy) is 91.40% and the mIoU (Mean Intersection over Union) is 85.02% on the validation set. The results indicate that the generation and enhancement of images with fewer defect categories in the collected data by using GAN can greatly improve the accuracy of defect recognition with fewer samples. After data augmentation, the DeepLabv3+ networks of PSPNet, U-Net and different backbone networks have been significantly improved, with an average increase of 20%~30% in mPA and mIoU. Adding the CBAM attention module can enhance the segmentation ability of U-Net and enable the model to have better pixel-level segmentation ability.

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

刘勇,李有淋.一种基于生成对抗网络和改进U-Net的排水管道缺陷识别模型[J].城市道桥与防洪,2026,(1):33-38.

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2025-09-20
  • 最后修改日期:2025-11-30
  • 录用日期:2025-11-30
  • 在线发布日期: 2026-01-18
  • 出版日期:
关闭