融合颜色保持与细节增强的煤矿井下图像去雾算法

A dehazing algorithm for underground coal mine images fusing color preservation and detail enhancement

  • 摘要: 煤矿井下采煤工作面环境复杂,作业过程中产生的煤尘和水雾等不均匀悬浮颗粒严重影响了监控图像的质量。图像处理中常出现去雾不彻底、过度增强和颜色失真等问题,而井下环境中缺乏配对的尘雾与清晰图像,限制了有监督去雾模型的发展。为实现有效的去雾处理,聚焦去雾后图像颜色保真与细节增强的核心问题,提出一种借助非配对数据的双分支融合去雾算法,由参数估计、颜色保持分支、细节增强分支和自适应感知融合四部分构成。首先,采用基于尘雾分布特征的参数估计方法获取初始透射率和大气光值,代入大气散射模型得到初始去雾图像;在颜色保持分支,将通道注意力机制嵌入到U型网络优化透射率,据此进行处理能提升去雾效果、并避免非浓雾区域的颜色失真;细节增强分支构建了包含双重注意力的残差网络架构,通过融合空间和通道信息增强关键特征的表达能力,引入非配对的清晰无雾图像进行对抗训练,提高去雾图像的细节表现力;在双分支融合阶段,兼顾色彩敏感区和纹理复杂区,采用自适应的加权融合策略,得到最终的去雾结果。为评估去雾算法的性能,采用真实的井下图像进行实验,并与几种典型的去雾算法进行对比。结果表明,提出的算法能有效去除尘雾的影响,减小颜色失真,提升图像的可视化效果。

     

    Abstract: The complex environment of underground coal mining faces seriously affects the quality of monitoring images due to uneven suspended particles such as coal dust and water mist generated during operation. Common problems in image processing include incomplete dehazing, over-enhancement, and color distortion. The lack of paired hazy and clear images in underground environments limits the development of supervised dehazing models. To achieve effective dehazing processing and focus on the core issues of color fidelity and detail enhancement in post-dehazing images, this paper proposes a dual-branch fusion dehazing algorithm using unpaired data, consisting of four components: parameter estimation, color preservation branch, detail enhancement branch, and adaptive perceptual fusion. First, targeting the non-uniform distribution characteristics of hazy conditions in underground coal mines, a parameter estimation method based on hazy distribution features is employed to obtain initial transmission and atmospheric light values, which are substituted into the atmospheric scattering model to generate initial dehazed images. In the color preservation branch, a channel attention mechanism is embedded into a U-shaped network to optimize transmission rates. This processing improves dehazing effectiveness and avoids color distortion in non-dense hazy areas. The detail enhancement branch constructs a residual network architecture with dual attention mechanisms, achieving collaborative preservation of local details and global structural information while fusing spatial and channel information to enhance the expression capability of key features. Unpaired clear haze-free images are introduced for adversarial training to improve the detail representation of dehazed images. In the dual-branch fusion stage, an adaptive weighted fusion strategy is adopted that considers both color-sensitive regions and texture-complex regions to obtain final dehazing results. In the experiments, 876 images with 1920×1080 resolution were collected from coal mining faces (412 clear images and 464 hazy images), with 404 hazy images used for training and 60 for testing. Data augmentation was performed through cropping to 512×512 image patches and random horizontal and vertical flipping to ensure training sample diversity. The collected hazy images exhibit obvious non-uniform characteristics, including various complex distribution scenarios such as dense haze and thin haze, which are distinctly different from traditional uniform hazy images. The overall network significantly improves the model's generalization ability and robustness in non-uniform hazy environments while maintaining detail enhancement through the introduction of adversarial training strategies. Quantitative comparative experimental results show that the proposed method achieves an information entropy of 5.52, FADE reduced to 0.49, PIQE reduced to 7.39, and an average running time of 1.304s on the test set. The comprehensive performance is superior to various typical dehazing algorithms, effectively removing non-uniform haze, reducing color distortion, and improving image visualization effects. Ablation experimental analysis shows that: removing the color preservation branch leads to decreased image brightness, color distortion, and reduced visual quality; the absence of the detail enhancement branch causes texture blurring and local structural loss; removing the Squeeze-and-Excitation (SE) attention module reduces the discriminative power of key feature channels; eliminating the dual attention network results in severe haze residue and insufficient detail recovery. This indicates that the synergistic effect of each module plays a key role in improving the clarity, color consistency, and texture integrity of dehazed images. In conclusion, the dual-branch fusion dehazing network proposed in this paper can achieve efficient dehazing with color fidelity and detail enhancement in complex underground coal mine environments, balancing the removal of non-uniform haze with visual quality improvement, providing reliable support for the stable operation of on-site monitoring systems.

     

/

返回文章
返回