Real-time tunnel lining leakage image semantic segmentation via multiple attention mechanisms

Abstract

One of the key objectives in tunnel illness detection is identifying tunnel lining leakage, and deep learning-based image semantic segmentation approaches can automatically locate tunnel lining leakage. However, in order to meet the real-time processing needs of professional mobile inspection equipment, existing leakage image segmentation approaches have difficulties in identifying real-time, dealing with voids, and dealing with edge discontinuities in the leaking zone. To address the aforementioned issues, this study introduces the PP-LiteSeg-Attn model, which takes the real-time semantic segmentation model PP-LiteSeg-B as baseline model, and combines the multi-layer CBAM attention mechanism and the CoT attention mechanism. Using the publically available dataset Water-Leakage, we trained and validated the PP-LiteSeg-Attn model, and attained IoU and F1 values of 88.18% and 93.72%, respectively, outperforming similar models in both measures. Extensive experiments show that the segmentation speed of the PP-LiteSeg-Attn model reaches 112.28FPS, which meets real-time requirements, and that the model can effectively solve problems such as the appearance of voids in the seepage area, discontinuity, and fuzzy segmentation of seepage edges. The PP-LiteSeg-Attn model is better applicable to complicated tunnel settings, offering technical references for real-time diagnosis of tunnel illnesses.