高级检索

基于半监督学习的双分支网络采煤工作面尘雾图像去雾算法

A dehazing algorithm for dust and fog images of coal mining face based on semi-supervised learning and dual-branch network

  • 摘要: 煤矿井下采煤工作面作业时产生煤尘、水雾等非均匀悬浮颗粒,视频图像出现模糊、细节丢失等降质现象。现有井下图像清晰化算法效果不佳,存在图像易失真、算法泛化性差等问题,提出了一种基于半监督学习的双分支网络去雾算法。该算法采用端到端的深度学习网络,将降质图像映射为清晰图像,并通过半监督学习方式进行训练。算法网络由去雾分支和细节修补分支构成:去雾分支为编码器解码器结构,编码器采用了中心差分卷积作为特征提取器,增强了高频信息的学习表征能力,解码器引入去噪模块,增强了网络去雾能力;细节修补分支由DCResBlock模块构成,利用扩张卷积增大感受野,引入少量参数解决去雾过程中的细节丢失问题。由于采煤工作面配对数据集难以收集,针对煤矿环境特点,改进了现有数据合成方法,将清晰图像合成为非均匀尘雾图像,并与真实尘雾图像一同作为训练样本对网络进行半监督训练。针对算法泛化性差的问题,引入了对比学习损失作为无监督损失,引导模型优化方向,提升对真实尘雾图像的去雾效果。为验证算法有效性,选取基于大气散射模型、YOLY、AECR-Net、CasDyF-Net、SSID 5种去雾算法对采煤工作面尘雾图像数据集进行实验,并采用相关指标与提出算法性能对比。结果表明,提出算法能更有效地降低尘雾密度,恢复细节信息,提高了煤矿井下采煤工作场景图像的视觉效果和图像质量,增强了其在工程应用中的实用性。

     

    Abstract: Coal mining operations in underground coal mine working faces generate non-uniform suspended particles such as coal dust and water mist, causing video images to suffer from degradation phenomena including blurring and detail loss. Existing underground image enhancement algorithms demonstrate limited effectiveness, suffering from image distortion and poor generalization capabilities. To address these issues, this paper proposes a semi-supervised dual-branch network dehazing algorithm. The proposed algorithm employs an end-to-end deep learning network to map degraded images to clear images, trained via a semi-supervised learning paradigm. The network architecture comprises a dehazing branch and a detail restoration branch: the dehazing branch follows an encoder-decoder structure, where the encoder utilizes central difference convolution as the feature extractor to enhance the learning and representation of high-frequency information, and the decoder incorporates a denoising module to strengthen the dehazing capability; the detail restoration branch is constructed with DCResBlock modules, leveraging dilated convolutions to enlarge the receptive field and introducing minimal parameters to address detail loss during the dehazing process. Due to the difficulty in collecting paired datasets from coal mining working faces, this study improves upon existing data synthesis methods based on the characteristics of coal mine environments, synthesizing clear images into non-uniform dust and haze images that serve as training samples alongside real dust and haze images for semi-supervised network training. To mitigate the issue of poor algorithm generalization, contrastive learning loss is introduced as an unsupervised loss to guide model optimization and enhance dehazing performance on real dust and haze images. To validate the effectiveness of the proposed algorithm, five dehazing algorithms—atmospheric scattering model-based method, YOLY, AECR-Net, CasDyF-Net, and SSID—are selected for comparative experiments on a coal mine working face dust and haze image dataset. Experimental results demonstrate that the proposed algorithm can more effectively reduce dust and haze density, restore detail information, improve the visual quality and image quality of underground coal mining working face scenes, and enhance its practicality for engineering applications.

     

/

返回文章
返回