Lightweight Image Super-Resolution with Adaptive Weighted Learning Network
About
Deep learning has been successfully applied to the single-image super-resolution (SISR) task with great performance in recent years. However, most convolutional neural network based SR models require heavy computation, which limit their real-world applications. In this work, a lightweight SR network, named Adaptive Weighted Super-Resolution Network (AWSRN), is proposed for SISR to address this issue. A novel local fusion block (LFB) is designed in AWSRN for efficient residual learning, which consists of stacked adaptive weighted residual units (AWRU) and a local residual fusion unit (LRFU). Moreover, an adaptive weighted multi-scale (AWMS) module is proposed to make full use of features in reconstruction layer. AWMS consists of several different scale convolutions, and the redundancy scale branch can be removed according to the contribution of adaptive weights in AWMS for lightweight network. The experimental results on the commonly used datasets show that the proposed lightweight AWSRN achieves superior performance on x2, x3, x4, and x8 scale factors to state-of-the-art methods with similar parameters and computational overhead. Code is avaliable at: https://github.com/ChaofWang/AWSRN
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Super-Resolution | Set5 | PSNR38.04 | 751 | |
| Super-Resolution | Urban100 | PSNR32.23 | 603 | |
| Super-Resolution | Set14 | PSNR33.66 | 586 | |
| Super-Resolution | B100 | PSNR32.21 | 418 | |
| Image Super-resolution | Urban100 x4 (test) | PSNR26.27 | 90 | |
| Image Super-resolution | Urban100 x2 (test) | PSNR32.23 | 72 | |
| Image Super-resolution | Urban100 x3 (test) | PSNR28.26 | 58 | |
| Image Super-resolution | Manga109 x2 (test) | PSNR38.66 | 52 | |
| Super-Resolution | Manga109 x3 (test) | PSNR33.64 | 49 | |
| Image Super-resolution | B100 x4 (test) | PSNR27.65 | 45 |