HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Selective Frequency Network for Image Restoration

{Alois Knoll Kai Huang Xiaochun Cao Xinwei Gao Wenqi Ren Zhenshan Bing Yi Tao Yuning Cui}

Abstract

Image restoration aims to reconstruct the latent sharp image from its corrupted counterpart. Besides dealing with this long-standing task in the spatial domain, a few approaches seek solutions in the frequency domain in consideration of the large discrepancy between spectra of sharp/degraded image pairs. However, these works commonly utilize transformation tools, e.g., wavelet transform, to split features into several frequency parts, which is not flexible enough to select the most informative frequency component to recover. In this paper, we exploit a multi-branch and content-aware module to decompose features into separate frequency subbands dynamically and locally, and then accentuate the useful ones via channel-wise attention weights. In addition, to handle large-scale degradation blurs, we propose an extremely simple decoupling and modulation module to enlarge the receptive field via global and window-based average pooling. Integrating two developed modules into a U-Net backbone, the proposed Selective Frequency Network (SFNet) performs favorably against state-of-the-art algorithms on five image restoration tasks, including single-image defocus deblurring, image dehazing, image motion deblurring, image desnowing, and image deraining.

Benchmarks

BenchmarkMethodologyMetrics
deblurring-on-goproSFNet
PSNR: 33.27
SSIM: 0.963
deblurring-on-rsblurSFNet
Average PSNR: 34.35
image-dehazing-on-sots-indoorSFNet
PSNR: 41.24
SSIM: 0.996
image-dehazing-on-sots-outdoorSFNet
PSNR: 40.05
SSIM: 0.996
single-image-deraining-on-rain100lSFNet
PSNR: 38.21
SSIM: 0.974

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