HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Revisiting Image Deblurring with an Efficient ConvNet

Lingyan Ruan Mojtaba Bemana Hans-peter Seidel Karol Myszkowski Bin Chen

Revisiting Image Deblurring with an Efficient ConvNet

Abstract

Image deblurring aims to recover the latent sharp image from its blurry counterpart and has a wide range of applications in computer vision. The Convolution Neural Networks (CNNs) have performed well in this domain for many years, and until recently an alternative network architecture, namely Transformer, has demonstrated even stronger performance. One can attribute its superiority to the multi-head self-attention (MHSA) mechanism, which offers a larger receptive field and better input content adaptability than CNNs. However, as MHSA demands high computational costs that grow quadratically with respect to the input resolution, it becomes impractical for high-resolution image deblurring tasks. In this work, we propose a unified lightweight CNN network that features a large effective receptive field (ERF) and demonstrates comparable or even better performance than Transformers while bearing less computational costs. Our key design is an efficient CNN block dubbed LaKD, equipped with a large kernel depth-wise convolution and spatial-channel mixing structure, attaining comparable or larger ERF than Transformers but with a smaller parameter scale. Specifically, we achieve +0.17dB / +0.43dB PSNR over the state-of-the-art Restormer on defocus / motion deblurring benchmark datasets with 32% fewer parameters and 39% fewer MACs. Extensive experiments demonstrate the superior performance of our network and the effectiveness of each module. Furthermore, we propose a compact and intuitive ERFMeter metric that quantitatively characterizes ERF, and shows a high correlation to the network performance. We hope this work can inspire the research community to further explore the pros and cons of CNN and Transformer architectures beyond image deblurring tasks.

Code Repositories

lingyanruan/lakdnet
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
deblurring-on-hide-trained-on-goproLaKDNet
PSNR (sRGB): 31.58
Params (M): 37.5
SSIM (sRGB): 0.946
deblurring-on-realblur-r-trained-on-goproLaKDNet
PSNR (sRGB): 36.08
SSIM (sRGB): 0.955
image-deblurring-on-goproLAKDNet
PSNR: 33.72
Params (M): 37.5
SSIM: 0.967

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
Revisiting Image Deblurring with an Efficient ConvNet | Papers | HyperAI