A Super Resolution Algorithm Based on Attention Mechanism and SRGAN Network

Autor: Baozhong Liu, Ji Chen
Jazyk: angličtina
Rok vydání: 2021
Předmět:
Zdroj: IEEE Access, Vol 9, Pp 139138-139145 (2021)
Druh dokumentu: article
ISSN: 2169-3536
DOI: 10.1109/ACCESS.2021.3100069
Popis: Image super-resolution reconstruction uses a specific algorithm to restore the low resolution blurred image in the same scene to a high resolution image. In recent years, with the vigorous development of deep learning, this technology has been widely used in many fields. In the field of image super-resolution reconstruction, more and more methods based on deep learning have been studied. According to the principle of GAN, a pseudo high-resolution image is generated by the generator, and then the discriminator calculates the difference between the image and the real high-resolution image to measure the authenticity of the image. Based on SRGAN (super resolution general adverse network), this paper mainly makes three improvements. First, it introduces the attention channel mechanism, that is, it adds Ca (channel attention) module to SRGAN network, and increases the network depth to better express high frequency features; Second, delete the original BN (batch normalization) layer to improve the network performance; Third, modify the loss function to reduce the impact of noise on the image. The experimental results show that the proposed method is superior to the current methods in both quantitative and qualitative indicators, and promotes the recovery of high-frequency detail information. The experimental results show that the proposed method improves the artifact problem and improves the PSNR (peak signal-to-noise ratio) on set5, set10 and bsd100 test sets.
Databáze: Directory of Open Access Journals