HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Hierarchical Conditional Flow: A Unified Framework for Image Super-Resolution and Image Rescaling

Liang Jingyun ; Lugmayr Andreas ; Zhang Kai ; Danelljan Martin ; Van Gool Luc ; Timofte Radu

Hierarchical Conditional Flow: A Unified Framework for Image
  Super-Resolution and Image Rescaling

Abstract

Normalizing flows have recently demonstrated promising results for low-levelvision tasks. For image super-resolution (SR), it learns to predict diversephoto-realistic high-resolution (HR) images from the low-resolution (LR) imagerather than learning a deterministic mapping. For image rescaling, it achieveshigh accuracy by jointly modelling the downscaling and upscaling processes.While existing approaches employ specialized techniques for these two tasks, weset out to unify them in a single formulation. In this paper, we propose thehierarchical conditional flow (HCFlow) as a unified framework for image SR andimage rescaling. More specifically, HCFlow learns a bijective mapping betweenHR and LR image pairs by modelling the distribution of the LR image and therest high-frequency component simultaneously. In particular, the high-frequencycomponent is conditional on the LR image in a hierarchical manner. To furtherenhance the performance, other losses such as perceptual loss and GAN loss arecombined with the commonly used negative log-likelihood loss in training.Extensive experiments on general image SR, face image SR and image rescalinghave demonstrated that the proposed HCFlow achieves state-of-the-artperformance in terms of both quantitative metrics and visual quality.

Code Repositories

jingyunliang/hcflow
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
image-rescaling-on-div2k-val-q30-4xHCFlow
PSNR: 25.89
SSIM: 0.6838
image-rescaling-on-div2k-val-q50-4xHCFlow
PSNR: 26.38
SSIM: 0.7029
image-rescaling-on-div2k-val-q70-4xHCFlow
PSNR: 26.79
SSIM: 0.7204
image-rescaling-on-div2k-val-q90-4xHCFlow
PSNR: 27.41
SSIM: 0.7485
video-super-resolution-on-msu-vsr-benchmarkHCFlow
1 - LPIPS: 0.923
ERQAv1.0: 0.713
FPS: 0.066
PSNR: 26.067
QRCRv1.0: 0
SSIM: 0.791
Subjective score: 4.262

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