Segmentation of tunnel water leakage based on a lightweight DeepLabV3+ model

Abstract The accurate and efficient detection of water leakage with complex backgrounds is crucial for the safety of metro operations. A lightweight segmentation method for metro tunnel water leakage based on transfer learning is proposed. Firstly, this is based on the Deeplabv3+ model and adopts MobileNetv3-Large as the backbone feature extraction network, which significantly reduces the network parameters and improves the detection speed; secondly, it incorporates the efficient channel attention mechanism, which enables the model to adaptively adjust the weights of the channel features and capture the inter-channel relationships in the image, which significantly improves the model’s ability for feature extraction ability; furthermore, for the problem of severe imbalance between positive and negative samples in the dataset, the recognition accuracy of complex samples is increased by optimizing the loss function; finally, the training method of transfer learning is utilized to solve the problem of scarcity of water leakage dataset, and to improve the model’s accuracy and generalization ability. The results show that the model has more significant detection accuracy and segmentation speed advantages than today’s mainstream semantic segmentation model. With strong generalization ability in complex environments (e.g. low illumination and multiple obstructions), model can be used for intelligent operation and maintenance in metro tunnel projects.