Image segmentation of tunnel water leakage defects in complex environments using an improved Unet model

Abstract

Computer vision technology provides an intelligent means for detecting tunnel water leakage areas. However, the accuracy of defect feature extraction and segmentation is limited by factors such as insufficient lighting and environmental interference inside tunnels. To address the problem, this paper proposes a tunnel water leakage area segmentation network model called Customized Side Guided-Unet (CSG-Unet), using Unet as the baseline model. The main contributions are: (1) To improve the accuracy of water leakage area extraction, a customized side guided term is introduced to direct the net’s attention to the changes in light and shade within the image. A parallel attention network module is designed to extract internal information from the guided term. Subsequently, a strengthened channel attention module aggregates the guided term and the original information to achieve accurate segmentation of water leakage areas; (2) To address the scarcity of tunnel water leakage area datasets, a basic dataset is constructed by collecting data from open-source datasets and manually gathered data in tunnels. On this basis, perspective transformation is used to change the camera viewpoint, gaussian noise is randomly added to the images in the dataset to simulate images taken in dimly lit scenes, thereby expanding the dataset and enhancing the network’s generalization. The CSG-Unet network was trained using the constructed training set, achieving a mean Intersection over Union (mi IoU) of 85.54%, a mean Dice coefficient (mi Dice) of 85.26%, and a mean Pixel Accuracy (mi PA) of 90.85%. Compared to its baseline network, U-Net (tiny), these metrics show an improvement of over 3.2% in each indicator. Finally, a visual comparison between the improved network and the baseline network further confirms that the proposed model can effectively adapt to the segmentation of water leakage areas in complex environments.