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

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

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

     

    Abstract: The complex environment of underground coal mining affects the quality of monitoring images due to uneven suspended particles generated during operation, such as coal dust and water mist. 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 nonuniform 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 nondense hazy areas. The detail enhancement branch has 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 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 and texture-complex regions to obtain the 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 by cropping 512×512 image patches and random horizontal and vertical flipping to ensure training sample diversity. The collected hazy images exhibit nonuniform characteristics, including complex distribution scenarios, such as dense haze and thin haze, which are different from traditional uniform hazy images. The overall network significantly improves the model’s generalization ability and robustness in nonuniform 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, fog aware density evaluator (FAED) reduced to 0.49, perception based image quality evaluator (PIQE) reduced to 7.39, and an average running time of 1.304 s on the test set. The comprehensive performance is superior to typical dehazing algorithms, effectively removing nonuniform 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 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. Furthermore, comparative experiments with different attention mechanisms validate the rationality of the proposed attention design, demonstrating its superior performance in modeling key features and enhancing dehazing stability under complex haze conditions. In conclusion, the proposed dual-branch fusion dehazing network can achieve efficient dehazing with color fidelity and detail enhancement in complex underground coal mine environments, balancing the removal of nonuniform haze with visual quality improvement to provide reliable support for the stable operation of on-site monitoring systems.

     

/

返回文章
返回