Journal of Xidian University ›› 2024, Vol. 51 ›› Issue (6): 171-181.doi: 10.19665/j.issn1001-2400.20240911

• Computer Science and Technology & Cyberspace Security • Previous Articles     Next Articles

Improved SwinIR for multi-feature fusion image super-resolution reconstruction

WANG Jinhua1(), WEI Ting1(), CAO Jie1,2,3(), CHEN Li3()   

  1. 1. School of Electrical Engineering and Information Engineering,Lanzhou University of Technology,Lanzhou 730050,China
    2. Gansu Manufacturing Information Engineering Research Center,Lanzhou 730050,China
    3. College of Information Engineering,Lanzhou City University,Lanzhou 730050,China
  • Received:2024-06-02 Online:2024-12-20 Published:2024-10-09

Abstract:

In the process of image super-resolution reconstruction based on the advanced SwinIR method,there is a problem that the local information modeling ability of low-resolution images is insufficient,resulting in inadequate feature extraction and poor quality of reconstructed images.An improved SwinIR multi-feature fusion image super-resolution reconstruction method is proposed.In the deep feature extraction module,the proposed algorithm first designs several series residual Swin Transformer blocks(RSTB),uses the Swin Transformer layer(STL) of RSTB for long-distance dependent modeling to extract high-frequency image information,and uses residual connections to achieve different levels of feature aggregation.Second,an alternating series spatial attention module and channel attention module(SA-CA) are designed to make up for the lack of local modeling ability of RSTB,so that the network can capture the missing context information on image space and channel dimension,and promote the reconstruction of edge details.Finally,the summation of shallow features and deep features is fused and transmitted to the reconstruction module for high-quality image reconstruction through along jump connection.Experimental results show that in the four test sets with magnifications of 2,3,and 4,the proposed improved algorithm achieves better results than SwinIR in terms of the peak signal-to-noise ratio and structural similarity,and the edge structure and overall contour of the reconstructed image are clearer in terms of visual effects.

Key words: image super-resolution reconstruction, swin transformer, spatial attention, channel attention, multi-feature fusion

CLC Number: 

  • TP391