HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

DifFace: Blind Face Restoration with Diffused Error Contraction

Yue Zongsheng ; Loy Chen Change

DifFace: Blind Face Restoration with Diffused Error Contraction

Abstract

While deep learning-based methods for blind face restoration have achievedunprecedented success, they still suffer from two major limitations. First,most of them deteriorate when facing complex degradations out of their trainingdata. Second, these methods require multiple constraints, e.g., fidelity,perceptual, and adversarial losses, which require laborious hyper-parametertuning to stabilize and balance their influences. In this work, we propose anovel method named DifFace that is capable of coping with unseen and complexdegradations more gracefully without complicated loss designs. The key of ourmethod is to establish a posterior distribution from the observed low-quality(LQ) image to its high-quality (HQ) counterpart. In particular, we design atransition distribution from the LQ image to the intermediate state of apre-trained diffusion model and then gradually transmit from this intermediatestate to the HQ target by recursively applying a pre-trained diffusion model.The transition distribution only relies on a restoration backbone that istrained with $L_2$ loss on some synthetic data, which favorably avoids thecumbersome training process in existing methods. Moreover, the transitiondistribution can contract the error of the restoration backbone and thus makesour method more robust to unknown degradations. Comprehensive experiments showthat DifFace is superior to current state-of-the-art methods, especially incases with severe degradations. Code and model are available athttps://github.com/zsyOAOA/DifFace.

Code Repositories

zsyoaoa/difface
Official
pytorch
Mentioned in GitHub
zsyoaoa/resshift
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
blind-face-restoration-on-celeba-testDFDNet
FID: 64.65
IDS: 86.21
LPIPS: 55.4
PSNR: 23.15
SSIM: 0.629
blind-face-restoration-on-celeba-testGFPGAN
FID: 46.99
IDS: 66.76
LPIPS: 49.5
PSNR: 22.18
SSIM: 0.631
blind-face-restoration-on-celeba-testPSFRGAN
FID: 52.14
IDS: 68.14
LPIPS: 50
PSNR: 22.74
SSIM: 0.63
blind-face-restoration-on-celeba-testGLEAN
FID: 60.3
IDS: 67.13
LPIPS: 46.9
PSNR: 23.41
SSIM: 0.666
blind-face-restoration-on-celeba-testDifFace
FID: 38.43
IDS: 62.39
LPIPS: 43.5
PSNR: 24.08
SSIM: 0.703
blind-face-restoration-on-celeba-testVQFR
FID: 45.84
IDS: 65.87
LPIPS: 47.1
PSNR: 21.94
SSIM: 0.585
blind-face-restoration-on-celeba-testPULSE
FID: 48.33
IDS: 74.97
LPIPS: 50.8
PSNR: 22.14
SSIM: 0.682

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