HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

Single Image Super-Resolution via a Holistic Attention Network

Ben Niu Weilei Wen Wenqi Ren Xiangde Zhang Lianping Yang Shuzhen Wang Kaihao Zhang Xiaochun Cao Haifeng Shen

Single Image Super-Resolution via a Holistic Attention Network

Abstract

Informative features play a crucial role in the single image super-resolution task. Channel attention has been demonstrated to be effective for preserving information-rich features in each layer. However, channel attention treats each convolution layer as a separate process that misses the correlation among different layers. To address this problem, we propose a new holistic attention network (HAN), which consists of a layer attention module (LAM) and a channel-spatial attention module (CSAM), to model the holistic interdependencies among layers, channels, and positions. Specifically, the proposed LAM adaptively emphasizes hierarchical features by considering correlations among layers. Meanwhile, CSAM learns the confidence at all the positions of each channel to selectively capture more informative features. Extensive experiments demonstrate that the proposed HAN performs favorably against the state-of-the-art single image super-resolution approaches.

Code Repositories

04RR/SOTA-Vision
pytorch
Mentioned in GitHub
wwlCape/HAN
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
image-super-resolution-on-bsd100-2x-upscalingHAN+
PSNR: 32.45
SSIM: 0.8431
image-super-resolution-on-bsd100-3x-upscalingHAN+
PSNR: 29.41
SSIM: 0.8116
image-super-resolution-on-bsd100-4x-upscalingHAN+
PSNR: 27.85
SSIM: 0.7454
image-super-resolution-on-bsd100-8x-upscalingHAN+
PSNR: 25.04
SSIM: 0.6075
image-super-resolution-on-manga109-2xHAN+
PSNR: 39.62
SSIM: 0.9787
image-super-resolution-on-manga109-3xHAN+
PSNR: 34.87
SSIM: 0.9509
image-super-resolution-on-manga109-4xHAN+
PSNR: 31.73
SSIM: 0.9207
image-super-resolution-on-manga109-8xHAN+
PSNR: 25.54
SSIM: 0.8080
image-super-resolution-on-set14-2x-upscalingHAN+
PSNR: 34.24
SSIM: 0.9224
image-super-resolution-on-set14-3x-upscalingHAN+
PSNR: 30.79
SSIM: 0.8487
image-super-resolution-on-set14-4x-upscalingHAN+
PSNR: 28.99
SSIM: 0.7907
image-super-resolution-on-set14-8x-upscalingHAN+
PSNR: 25.39
SSIM: 0.6552
image-super-resolution-on-set5-2x-upscalingHAN+
PSNR: 38.33
SSIM: 0.9299
image-super-resolution-on-set5-3x-upscalingHAN+
PSNR: 34.85
SSIM: 0.9300
image-super-resolution-on-set5-8x-upscalingHAN+
PSNR: 27.47
SSIM: 0.7920
image-super-resolution-on-urban100-2xHAN+
PSNR: 33.53
SSIM: 0.9398
image-super-resolution-on-urban100-3xHAN+
PSNR: 29.21
SSIM: 0.8710
image-super-resolution-on-urban100-4xHAN+
PSNR: 27.02
SSIM: 0.8131
image-super-resolution-on-urban100-8xHAN+
PSNR: 23.20
SSIM: 0.6518

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