Image Reconstruction Algorithm Based on Improved Super-Resolution Generative Adversarial Network

被引:4
作者
Zha Tibo [1 ]
Luo Lin [1 ]
Yang Kai [1 ]
Zhang Yu [1 ]
Li Jinlong [1 ]
机构
[1] Southwest Jiaotong Univ, Sch Phys Sci & Technol, Chengdu 610031, Sichuan, Peoples R China
关键词
image processing; super-resolution; deep learning; residual network; generative adversarial network;
D O I
10.3788/LOP202158.0810005
中图分类号
TM [电工技术]; TN [电子技术、通信技术];
学科分类号
0808 ; 0809 ;
摘要
Aiming at the problem that the existing pixel loss-based super-resolution image reconstruction algorithms have poor reconstruction effect on high-frequency details, such as textures, an image reconstruction algorithm based on an improved super-resolution generative adversarial network (SRGAN) is proposed in this paper. First, remove the batch normalization layers in the generator, combine the multi-level residual network and dense connections, and use the residual-in-residual dense blocks to improve the network's ability for feature extraction. Then, the mean square error and perceptual loss are combined as the loss function to guide the generator training, which preserves the image's high-frequency details and avoids the artifacts' appearance. Finally, the last Sigmoid layer of the discriminator is removed to better converge the training process, and the relativistic loss function is used to guide the discriminator training. The experimental results on the COCO dataset show that compared with the original SRGAN algorithm, the peak signal-to-noise ratio (PSNR) and structural similarity (SSIM) of the algorithm in the Set5 data set are increased by 0.86 dB and 0.0123, respectively, in the Set14 data set, the PSNR and SSIM of the algorithm are improved by 0.69 dB and 0.0090, respectively. The mean opinion index and visual effect of the algorithm are far better than other algorithms.
引用
收藏
页数:11
相关论文
共 21 条
[21]  
Zhang Shufang, 2015, Computer Engineering and Applications, V51, P13, DOI 10.3778/j.issn.1002-8331.1507-0044