基于深度学习的消落带岩体裂隙识别方法研究

    Study on a Deep Learning-Based Method for Rock Fracture Identification in the Water-Level Fluctuation Zone

    • 摘要: 针对三峡库区消落带岩体裂隙在低对比度、复杂背景条件下识别困难,以及UNet模型参数量大、计算开销高、难以满足边缘设备部署需求的问题,本文提出了一种轻量化裂隙语义分割模型LiteUNet。该模型在UNet基础上引入深度可分离卷积、多尺度上下文感知模块、轻量解码器及深度监督机制,以提升对细小裂隙和复杂边界的识别能力。基于巫山段消落带岩体裂隙图像构建数据集,并通过数据增强扩充样本规模,在统一实验条件下与多种主流语义分割模型进行了对比分析。结果表明,LiteUNet的mIoU和F1值分别达到82.92%和90.66%,参数量和计算量分别仅为2.76 M和17.61 GFLOPs,显著低于主流语义分割模型,且在现场应用中能够较稳定地提取消落带岩体裂隙信息,在保证识别精度的同时兼顾了运算效率,适用于复杂消落带场景下的裂隙快速识别与边缘部署。

       

      Abstract: To address the difficulty of rock fracture detection in the water-level fluctuation zone of the Three Gorges Reservoir under low contrast and complex background conditions, as well as the large number of parameters and high computational cost of the UNet model that limit its deployment on edge devices, a lightweight semantic segmentation model, LiteUNet, is proposed. Based on the UNet architecture, depthwise separable convolution, a multi-scale context module, a lightweight decoder, and a deep supervision mechanism are introduced to improve the detection of fine fractures and complex boundaries. A dataset of rock fracture images from the Wushan section of the fluctuation zone was constructed, and data augmentation was applied to expand the sample size. The proposed model was compared with several mainstream semantic segmentation models under the same experimental conditions. The results show that LiteUNet achieves an mIoU of 82.92% and an F1-score of 90.66%. The number of parameters and computational cost are reduced to 2.76 M and 17.61 GFLOPs, respectively, which are significantly lower than those of mainstream models. In practical applications, fracture information in the fluctuation zone can be stably extracted, and a good balance between accuracy and efficiency is achieved. Therefore, the proposed model is suitable for fast fracture detection and deployment on edge devices in complex fluctuation zone scenarios.

       

    /

    返回文章
    返回