The paper proposes a two-stage image dehazing framework called DehazeDDPM that combines physical modeling and diffusion models to tackle the challenging dense-haze image dehazing task.
In the first stage, the physical modeling network estimates the transmission map, haze-free image, and atmospheric light based on the Atmospheric Scattering Model (ASM). This pulls the distribution of the hazy data closer to that of the clear data and endows DehazeDDPM with fog-aware ability.
In the second stage, the conditional DDPM exploits its strong generation ability to compensate for the haze-induced information loss, working in conjunction with the physical modeling. The transmission map from the first stage is used as a confidence map to guide the learning of the second stage, mitigating the difficulty of DDPM for image dehazing.
Additionally, a frequency prior optimization strategy is introduced to better recover the high-frequency details. Extensive experiments demonstrate that DehazeDDPM achieves state-of-the-art performance on both synthetic and real-world hazy datasets, with much better perceptual quality on complex real-world scenes.
In un'altra lingua
dal contenuto originale
arxiv.org
Approfondimenti chiave tratti da
by Hu Yu,Jie Hu... alle arxiv.org 04-16-2024
https://arxiv.org/pdf/2308.11949.pdfDomande più approfondite