基于改进TransUNet的隧道衬砌渗漏水语义分割研究

Semantic Segmentation of Water Leakage in Tunnel Linings Based on an Improved TransUNet

  • 摘要: 传统人工检测隧道衬砌渗漏水存在主观性强、效率低等不足,现有深度学习模型在复杂场景下识别精度有限,为提升隧道渗漏水检测的准确性与效率,构建一种适用于复杂背景下的渗漏水智能识别模型,将Vision Transformer(ViT)融入UNet 架构形成TransUNet网络,结合SE-Block通道注意力机制对全局语义特征进行加权,并优化编码器与解码器间的跳跃连接,构建 SE-TransUNet 模型。基于混合渗漏水数据集对模型进行训练,利用消融实验验证SE-Block与ViT模块的有效性,结合 Score-CAM热力图分析模型对渗漏水特征的关注机制。结果表明:(1) 相较于8种主流语义分割模型,SE-TransUNet综合性能较好,其交并比、召回率、精确率、准确率、F1-Score分别达0.832 5、0.954 1、0.867 4、0.948 7、0.908 2,对凹痕阴影、污渍干扰及浅渗水印迹等场景具有较强适应性;(2)消融实验证实,SE模块与ViT模块对模型性能提升具有协同增益作用,基线模型IoU较移除全部SE模块的变体模型提升0.0461,较同时移除SE模块和ViT模块的变体模型提升0.0711;(3)Score-CAM 热力图及删除、插入曲线结果显示,SE模块可扩大模型对渗漏水区域的关注范围,增强特征连续性,显著提升复杂环境下的抗干扰能力。研究表明所构建SE-TransUNet模型在本文数据集与实验设置下具有较好的渗漏水识别能力,可为隧道衬砌病害智能检测研究提供参考。

     

    Abstract: Traditional manual inspection of water leakage in tunnel linings is highly subjective and inefficient, while existing deep learning models still have limited recognition accuracy in complex scenarios. To improve the accuracy and efficiency of water-leakage detection in tunnel linings, this study develops an intelligent water-leakage recognition model for complex backgrounds. A Vision Transformer (ViT) was incorporated into the UNet architecture to form a TransUNet network. A squeeze-and-excitation block (SE-Block) channel attention mechanism was introduced to weight global semantic features, and the skip connections between the encoder and decoder were optimized, thereby constructing the SE-TransUNet model. The model was trained on a mixed water-leakage dataset. Ablation experiments were conducted to verify the effectiveness of the SE-Block and ViT modules, and score-weighted class activation mapping (Score-CAM) heatmaps were used to analyze the model’s attention mechanism for water-leakage features. The results show that: (1) compared with eight mainstream semantic segmentation models, SE-TransUNet achieves better overall performance, with intersection over union (IoU), recall, precision, accuracy, and F1-score values of 0.8325, 0.9541, 0.8674, 0.9487, and 0.9082, respectively, and shows strong adaptability to scenarios involving indentation shadows, stain interference, and faint water-leakage traces; (2) the ablation results confirm that introducing the SE and ViT modules has a positive synergistic effect on model performance, with the IoU of the baseline model being 0.0461 and 0.0711 higher than those of the variants without all SE modules and without both SE and ViT modules, respectively; and (3) the Score-CAM heatmaps and deletion/insertion curves indicate that the SE module helps expand the model’s attention coverage over water-leakage regions, enhance feature continuity, and improve resistance to interference in complex environments. The proposed SE-TransUNet model demonstrates effective water-leakage recognition capability with the dataset and experimental settings used in this study, providing a reference for intelligent detection of tunnel lining defects.

     

/

返回文章
返回