吉林大学学报(理学版) ›› 2021, Vol. 59 ›› Issue (6): 1491-1498.

• • 上一篇    下一篇

基于生成对抗网络的单图像超分辨率重建

朱海琦1, 李宏1, 李定文1, 李富2   

  1. 1. 东北石油大学 电气信息工程学院, 黑龙江 大庆 163318; 2. 大庆钻探工程公司 钻井一公司, 黑龙江 大庆 163458
  • 收稿日期:2020-07-10 出版日期:2021-11-26 发布日期:2021-11-26
  • 通讯作者: 李宏 E-mail:853386766@qq.com

Single Image Super-Resolution Reconstruction Based on Generative Adversarial Network

ZHU Haiqi1, LI Hong1, LI Dingwen1, LI Fu2   

  1. 1. School of Electrical Information Engineering, Northeast Petroleum University, Daqing 163318, Heilongjiang Province, China;  2. Drilling No.1 Company, Daqing Drilling Engineering Company, Daqing 163458, Heilongjiang Province, China
  • Received:2020-07-10 Online:2021-11-26 Published:2021-11-26

摘要: 针对当前卷积神经网络未能充分利用浅层特征信息, 并难以捕获各特征通道间的依赖关系、 丢失高频信息的问题, 提出一种新的生成对抗网络用于图像超分辨率重建. 首先, 在生成器中引入WDSR-B残差块充分提取浅层特征信息; 其次, 将GCNet模块和像素注意力机制相结合加入到生成器和鉴别器中, 学习各特征通道的重要程度和高频信息; 最后, 采用谱归一化代替不利于图像超分辨率的批规范化, 减少计算开销, 稳定训练. 实验结果表明, 该算法与其他经典算法相比能有效提高浅层特征信息的利用率, 较好地重建出图像的细节信息和几何特征, 提高超分辨率图像的质量.

关键词: 图像超分辨率, 生成对抗网络, 注意力机制, 残差网络

Abstract: Aiming at the problem that the current convolutional neural network could not make full use of the shallow feature information, 
and it was difficult to capture the dependency between each feature channel, resulting in the loss of high-frequency information, we proposed a new generative adversarial network for image super-resolution reconstruction. Firstly, the WDSR-B residual block was introduced into the generator to fully extract the shallow feature information. Secondly, the GCNet module and pixel attention mechanism were combined into the generator and discriminator to learn the importance and high-frequency information of each feature channel. Finally, using spectral normalization instead of batch normalization that was not conducive to image super-resolution, and reduced computational overhead and stabilize training. Experimental results show that compared with other classical algorithms, the proposed algorithm can effectively improve the utilization of shallow feature information, better reconstruct the detailed information and geometric features of the image, and improve the quality of super-resolution images.

Key words:  , image super-resolution, generative adversarial network, attention mechanism, residual network

中图分类号: 

  • TP391