← all repositories

Algolzw/image-restoration-sde

PyTorch implementation of diffusion models using mean-reverting stochastic differential equations for comprehensive image restoration tasks.

717 stars Python Image · Video · Audio
image-restoration-sde
Velocity · 7d
+0.6
★ / day
Trend
steady
star history

This repository provides official implementations of IR-SDE and Refusion, two diffusion-based image restoration methods. IR-SDE uses mean-reverting stochastic differential equations to model the restoration process, while Refusion enables large-size realistic image restoration using latent-space diffusion models. Both approaches handle tasks including deblurring, denoising, super-resolution, deraining, dehazing, inpainting, and shadow removal. The methods are trained on synthetic and real-world datasets and achieved state-of-the-art results, winning the NTIRE 2023 Image Shadow Removal Challenge.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.