arrow
Return

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

delete2024-01-01
delete15
delete
OA
AI
X
Xuechao Zou
K
Kai Li
兴军亮 (Junliang Xing)
Y
Yu Zhang
S
Shiying Wang
L
Lei Jin
P
Pin Tao *
DOI:10.1109/TGRS.2024.3365806delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
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. Recent advances in deep learning-based cloud removal methods have been significant, but the quality of image generation still needs to be improved. Diffusion models have demonstrated remarkable success in diverse image generation tasks, showcasing their potential in addressing this challenge. This article 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 (TCFBlock) 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 three commonly used benchmark datasets demonstrate that DiffCR consistently achieves the state-of-the-art (SOTA) 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, pretrained models, and all the experimental results will be publicly available at https://github.com/XavierJiezou/DiffCR upon the paper's acceptance of this work.
Keywords:
Autoencoder
cloud removal
deep learning
diffusion models
optical satellite images

Journal

IEEE Transactions on Geoscience and Remote Sensing cover
IEEE Transactions on Geoscience and Remote Sensing
IF:
8.6
Papers:
2.1W
Citations:
10.7W

Organization

B
beijing university of posts & telecommunications
Scholars:
1.4W
Papers: 1.2W
Citations: 9
T
tsinghua university
Scholars:
11.7W
Papers: 10.0W
Citations: 137
Q
Qinghai University
Scholars:
6.1K
Papers: 3.3K
Citations: 4.4K
M
ministry of education - china
Scholars:
2.5W
Papers: 1.0W
Citations: 13
researcher View more organizations