用混合模型提升卫星图像去雾效果,保持细节与结构清晰
Satellite Image Utilization for Dehazing with Swin Transformer-Hybrid U-Net and Watershed loss
- 结合Swin Transformer与U-Net,兼顾全局上下文与局部细节
- 在RICE数据集上达33.24 dB PSNR和0.967 SSIM,优于现有方法
- 适合遥感、环境监测等需高精度卫星图像的场景
卫星影像在多个领域至关重要,但大气干扰和雾霾会严重降低图像清晰度,影响信息提取精度。为此,本文提出一种融合Swin Transformer与U-Net的混合去雾框架SUFE-RNOBWA,通过在编码器和解码器中使用基于Swin Transformer的残差嵌套密集块(SwinRRDB),有效提取全局上下文与精细空间结构特征,有助于卫星图像的结构保持。同时引入包含L2损失、引导损失和新型分水岭损失(watershed loss)的复合损失函数,增强边缘结构保留与像素级精度。该架构在多种大气条件下均表现鲁棒,恢复图像结构一致性高。实验表明,所提方法在RICE和SateHaze1K数据集上优于现有先进模型。尤其在RICE数据集上,达到33.24 dB PSNR和0.967 SSIM,显著提升去雾性能。本研究为缓解卫星影像大气干扰提供了有效方案,具有广泛遥感应用潜力。
原文摘要 · Abstract (English)
Satellite imagery plays a crucial role in various fields; however, atmospheric interference and haze significantly degrade image clarity and reduce the accuracy of information extraction. To address these challenges, this paper proposes a hybrid dehazing framework that integrates Swin Transformer and U-Net to balance global context learning and local detail restoration, called SUFERNOBWA. The proposed network employs SwinRRDB, a Swin Transformer-based Residual-in-Residual Dense Block, in both the encoder and decoder to effectively extract features. This module enables the joint learning of global contextual information and fine spatial structures, which is crucial for structural preservation in satellite image. Furthermore, we introduce a composite loss function that combines L2 loss, guided loss, and a novel watershed loss, which enhances structural boundary preservation and ensures pixel-level accuracy. This architecture enables robust dehazing under diverse atmospheric conditions while maintaining structural consistency across restored images. Experimental results demonstrate that the proposed method outperforms state-of-the-art models on both the RICE and SateHaze1K datasets. Specifically, on the RICE dataset, the proposed approach achieved a PSNR of 33.24 dB and an SSIM of 0.967, which is a significant improvement over existing method. This study provides an effective solution for mitigating atmospheric interference in satellite imagery and highlights its potential applicability across diverse remote sensing applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。