arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2308.04417cs.CVcs.LGeess.IV
收藏

DiffCR:一种用于光学卫星影像去云的快速条件扩散框架

DiffCR: A Fast Conditional Diffusion Framework for Cloud Removal from Optical Satellite Images

Xuechao Zou, Kai Li, Junliang Xing, Yu Zhang, Shiying Wang, Lei Jin, Pin Tao

AI总结:

本文提出DiffCR条件扩散框架,通过解耦编码器和高效时间与条件融合模块实现光学卫星影像高性能去云,以极低参数量和计算复杂度在多个基准上取得最优性能。

摘要:

光学卫星图像是一种关键数据源;然而,云覆盖常常损害其质量,阻碍图像应用与分析。因此,有效去除光学卫星图像中的云已成为一个突出的研究方向。尽管近年来云去除的进展主要依赖生成对抗网络,但这类方法可能产生次优的图像质量,而扩散模型已在多种图像生成任务中展现出显著成功,显示出其应对这一挑战的潜力。本文提出一个名为DiffCR的新框架,该框架利用条件引导扩散与深度卷积网络实现光学卫星影像的高性能云去除。具体而言,我们引入一个解耦编码器用于条件图像特征提取,提供稳健的颜色表示,以确保条件输入与合成输出之间外观信息的高度相似性。此外,我们在云去除模型中提出一种新颖且高效的时间与条件融合模块,以较低计算成本准确模拟条件图像外观与目标图像之间的对应关系。在两个常用基准数据集上的大量实验评估表明,DiffCR在所有指标上持续取得最先进性能,其参数量和计算复杂度分别仅为先前最佳方法的5.1%和5.4%。源代码、预训练模型及所有实验结果将在论文被接收后于https://github.com/XavierJiezou/DiffCR公开。

英文摘要:

Optical satellite images are a critical data source; however, cloud cover often compromises their quality, hindering image applications and analysis. Consequently, effectively removing clouds from optical satellite images has emerged as a prominent research direction. While recent advancements in cloud removal primarily rely on generative adversarial networks, which may yield suboptimal image quality, diffusion models have demonstrated remarkable success in diverse image-generation tasks, showcasing their potential in addressing this challenge. This paper presents a novel framework called DiffCR, which leverages conditional guided diffusion with deep convolutional networks for high-performance cloud removal for optical satellite imagery. Specifically, we introduce a decoupled encoder for conditional image feature extraction, providing a robust color representation to ensure the close similarity of appearance information between the conditional input and the synthesized output. Moreover, we propose a novel and efficient time and condition fusion block within the cloud removal model to accurately simulate the correspondence between the appearance in the conditional image and the target image at a low computational cost. Extensive experimental evaluations on two commonly used benchmark datasets demonstrate that DiffCR consistently achieves state-of-the-art performance on all metrics, with parameter and computational complexities amounting to only 5.1% and 5.4%, respectively, of those previous best methods. The source code, pre-trained models, and all the experimental results will be publicly available at https://github.com/XavierJiezou/DiffCR upon the paper's acceptance of this work.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑