Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Emphasizing Crucial Features for Efficient Image Restoration

About

Image restoration is a challenging ill-posed problem which estimates latent sharp image from its degraded counterpart. Although the existing methods have achieved promising performance by designing novelty architecture of module, they ignore the fact that different regions in a corrupted image undergo varying degrees of degradation. In this paper, we propose an efficient and effective framework to adapt to varying degrees of degradation across different regions for image restoration. Specifically, we design a spatial and frequency attention mechanism (SFAM) to emphasize crucial features for restoration. SFAM consists of two modules: the spatial domain attention module (SDAM) and the frequency domain attention module (FDAM). The SFAM discerns the degradation location through spatial selective attention and channel selective attention in the spatial domain, while the FDAM enhances high-frequency signals to amplify the disparities between sharp and degraded image pairs in the spectral domain. Additionally, to capture global range information, we introduce a multi-scale block (MSBlock) that consists of three scale branches, each containing multiple simplified channel attention blocks (SCABlocks) and a multi-scale feed-forward block (MSFBlock). Finally, we propose our ECFNet, which integrates the aforementioned components into a U-shaped backbone for recovering high-quality images. Extensive experimental results demonstrate the effectiveness of ECFNet, outperforming state-of-the-art (SOTA) methods on both synthetic and real-world datasets.

Hu Gao, Bowen Ma, Ying Zhang, Jingfan Yang, Jing Yang, Depeng Dang• 2024

Related benchmarks

TaskDatasetResultRank
Image DenoisingSIDD
PSNR24.33
124
Multi-task Image RestorationAll-in-One IR Benchmark Average
PSNR32.2
31
DerainingAIO-IR Deraining
PSNR30.44
21
DesnowingAIO-IR Desnowing
PSNR32.54
20
DeblurringAIO-IR Deblurring
PSNR27.58
20
Underwater Image EnhancementUIEB C60
UIQM2.517
18
Image DenoisingAll-in-One IR Benchmark Denoising Noise Level 15
PSNR33.33
15
Image DehazingAll-in-One IR Benchmark Dehazing
PSNR37.09
15
Underwater Image EnhancementUIEB
SSIM0.856
14
Image RestorationRealRain 1k-L
PSNR23.69
10
Showing 10 of 11 rows

Other info

Follow for update