弱光条件下基于改进Unet模型的隧道渗水病害图像分割

Improved UNet Model-based Image Segmentation for Tunnel Seepage Defects under Low-light Conditions

  • 摘要: 计算机视觉技术为隧道渗水区域检测提供了一种智能化手段,但受隧道内光线不足以及管线遮挡等复杂环境影响,病害特征提取和分割的精确度受限。鉴于此,以Unet为基线模型,提出一种隧道渗水区域分割网络Light-Unet模型。具体包括:(1)针对隧道内渗水区域提取准确性低的问题,在基线模型中引入光影变换引导项,并设计平行注意力模块提取引导项内部信息,通过加强通道注意力模块实现对引导项及网络深层信息的深度聚合;(2)针对隧道渗水区域数据集稀少的问题,在某施工隧道人工采集数据集,并运用透视变换改变相机视角,以扩充数据集;(3)针对隧道中光线不足的问题,在数据集的图像中随机添加高斯噪声与椒盐噪声,来模拟昏暗场景下拍摄的图像,增强网络的泛化性。最终,运用构建的训练集训练 Light-Unet 网络,网络的 mIoU 为 88.84%,mDice 为88.04%,均显著高于基线模型。以可视化的形式对比改进网络与基线网络,结果表明,改进网络能够更好地适应隧道内复杂环境下(弱光及线路遮挡)的渗水区域分割。

     

    Abstract: Computer vision technology offers an intelligent solution for tunnel seepage area detection, yet its feature extraction and segmentation accuracy remains constrained by low-light conditions and pipeline obstructions. This study proposes a Light-UNet model based on the Unet model, an enhanced tunnel seepage zone segmentation network with three key contributions: (1) To enhance the extraction accuracy of the seepage area, a light-shadow transformation boot item is introduced, and a parallel attention module is designed to extract internal information of the boot item. Subsequently, channel attention mechanisms are employed to aggregate the boot item with deep network information. (2) To overcome the scarcity of tunnel seepage datasets, the data was manually collected from a tunnel under construction. Perspective transformation was used to alter camera angles and expand the dataset. (3) To address insufficient lighting in tunnels, Gaussian noise and salt-and-pepper noise were randomly added to the dataset images to simulate images captured in dim environments, thereby enhancing the network’s generalization ability. Ultimately, the Light-Unet network was trained using the constructed training set, achieving a mean Intersection over Union (mIoU) of 88.84% and a mean Dice coefficient (mDice) of 88.04%, both significantly higher than the Unet baseline model. Visualization comparisons between the improved and baseline networks demonstrate that the improved network better adapts to the segmentation of seepage areas in complex environments, including poor lighting and pipeline obstructions.

     

/

返回文章
返回