Li Baoxian, Chu Xu, Lin Fusheng, Wu Fengyuan, Jin Shuo, Zhang Kexin
School of Transportation and Geomatics Engineering, Shenyang Jian Zhu University, Shenyang, 110168, China.
China Railway Shanghai Engineering Group Co., ltd, Shanghai, China.
Sci Rep. 2024 Nov 15;14(1):28234. doi: 10.1038/s41598-024-79919-6.
The accurate detection of tunnel lining cracks and prompt identification of their primary causes are critical for maintaining tunnel availability. The advancement of deep learning, particularly in the domain of convolutional neural network (CNN) for image segmentation, has made tunnel lining crack detection more feasible. However, the CNN-based technique for tunnel lining crack detection commonly prioritizes increasing algorithmic complexity to enhance detection accuracy, posing a challenge in balancing the accuracy of detection and the efficiency of the algorithm. Motivated by the superior performance of Unet in image segmentation, this paper proposes a lightweight tunnel lining crack detection model named Mini-Unet, which refined the Unet architecture and utilized depthwise separable convolutions (DSConv) to replace some standard convolution layers. In the optimization of the proposed model parameters, applying a hybrid loss function that integrated dice loss and cross-entropy loss effectively tackled the imbalance between crack and background categories. Several models were set up to contrast with Mini-Unet and the experimental results were analyzed. Mini-Unet achieves a mean intersection over union (MIoU) of 60.76%, a mean precision of 84.18%, and a frame per second (FPS) of 5.635, respectively. Mini-Unet outperforms several mainstream models, enabling rapid detection while maintaining identified accuracy and facilitating the practical application of AI power for real-time tunnel lining crack detection.
准确检测隧道衬砌裂缝并迅速识别其主要成因对于维持隧道的可用性至关重要。深度学习的发展,特别是在用于图像分割的卷积神经网络(CNN)领域,使得隧道衬砌裂缝检测变得更加可行。然而,基于CNN的隧道衬砌裂缝检测技术通常优先增加算法复杂度以提高检测精度,这在平衡检测精度和算法效率方面带来了挑战。受Unet在图像分割方面的卓越性能启发,本文提出了一种名为Mini-Unet的轻量级隧道衬砌裂缝检测模型,该模型对Unet架构进行了优化,并利用深度可分离卷积(DSConv)替换了一些标准卷积层。在优化所提出模型的参数时,应用结合了骰子损失和交叉熵损失的混合损失函数有效解决了裂缝与背景类别之间的不平衡问题。设置了几个模型与Mini-Unet进行对比并分析了实验结果。Mini-Unet分别实现了60.76%的平均交并比(MIoU)、84.18%的平均精度以及每秒5.635帧(FPS)。Mini-Unet优于几个主流模型,能够在保持识别精度的同时实现快速检测,并促进人工智能在实时隧道衬砌裂缝检测中的实际应用。