A lightweight metro tunnel water leakage identification algorithm via machine vision

• Published in: Engineering failure analysis 2023-08, Vol.150, p.107327, Article 107327
• Main Authors: Tan, Lei, Hu, Xiaoxi, Tang, Tao, Yuan, Dajun
• Format: Article
• Language: English
• Subjects: Image cropping; Image segmentation; Image stitching; Lightweight segmentation network; ShuffleNet v2; Skip-connection; Water leakage segmentation
• ISSN: 1350-6307; 1873-1961
• DOI: 10.1016/j.engfailanal.2023.107327

The existing methods of metro tunnel water leakage segmentation tasks are time-consuming, which is not easy for practical applications. In this paper, we propose a lightweight tunnel water leakage segmentation algorithm, covering image process and water leakage segmentation. The huge image collected by sensing vehicle is too large to be processed directly, so we put forward an image cropping and stitching algorithm. Meanwhile, we focus on the point of inference speed and thus design a Lightweight Segmentation Network (LSNet) for metro shield tunnel water leakage. More precisely, we use the ShuffleNet v2 as the encoder to accelerate the network. Also, we build a decoder using the skip-connection structure to maintain accuracy. Experiments based on a real dataset show the superiority of the algorithm proposed. The proposed model reaches 123.68 Frames Per Second (FPS) after the acceleration of TensorRT. •An LSNet for tunnel water leakage detection with 123.68 FPS is proposed.•The skip-connection reduces information loss of down-sampling in the decoder.•The image cropping and stitching help LSNet under the large image.•Sufficient experiments on real images are made to prove the superiority of the LSNet.