HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

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

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

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

Abstract

Optical satellite images are a critical data source; however, cloud coveroften compromises their quality, hindering image applications and analysis.Consequently, effectively removing clouds from optical satellite images hasemerged as a prominent research direction. While recent advancements in cloudremoval primarily rely on generative adversarial networks, which may yieldsuboptimal image quality, diffusion models have demonstrated remarkable successin diverse image-generation tasks, showcasing their potential in addressingthis challenge. This paper presents a novel framework called DiffCR, whichleverages conditional guided diffusion with deep convolutional networks forhigh-performance cloud removal for optical satellite imagery. Specifically, weintroduce a decoupled encoder for conditional image feature extraction,providing a robust color representation to ensure the close similarity ofappearance information between the conditional input and the synthesizedoutput. Moreover, we propose a novel and efficient time and condition fusionblock within the cloud removal model to accurately simulate the correspondencebetween the appearance in the conditional image and the target image at a lowcomputational cost. Extensive experimental evaluations on two commonly usedbenchmark datasets demonstrate that DiffCR consistently achievesstate-of-the-art performance on all metrics, with parameter and computationalcomplexities amounting to only 5.1% and 5.4%, respectively, of those previousbest methods. The source code, pre-trained models, and all the experimentalresults will be publicly available at https://github.com/XavierJiezou/DiffCRupon the paper's acceptance of this work.

Code Repositories

xavierjiezou/diffcr
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
cloud-removal-on-sen12ms-crDiffCR
MAE: 0.019
PSNR: 31.77
SAM: 5.821
SSIM: 0.902

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
DiffCR: A Fast Conditional Diffusion Framework for Cloud Removal from Optical Satellite Images | Papers | HyperAI