基于双注意力机制信息蒸馏网络的图像超分辨率复原算法

针对超分辨率复原技术中网络层数不断加深导致的网络训练困难、特征信息利用率低等问题,设计并实现了一种基于双注意力的信息蒸馏网络(IDN)的图像超分辨率复原算法.首先,利用IDN较低的计算复杂度及信息蒸馏模块提取更多特征的优势,通过引入残差注意力模块(RAM)并考虑图像通道之间的相互依赖性来自适应地重新调整特征权重,从而进...

Image super-resolution restoration algorithm based on information distillation network with dual attention mechanism

Aiming at the problems of network training difficulty and low utilization rate of feature information caused by increasing network layers in super-resolution restoration technology， an image super-resolution restoration algorithm based on dual attention Information Distillation Network （IDN） was designed and implemented. Firstly， by taking the advantage of the low computational complexity of IDN and the advantage of the information distillation module by which more features were extracted， the weights of the features were readjust adaptively by introducing the Residual Attention Module （RAM） and considering the interdependence of image channels， so as to further improve the reconstruction ability of high-resolution details of images. Then， a new mixed loss function sensitive to edge information was designed to refine the image and accelerate the convergence of the network. Test results on Set5， Set14， BSD100 and Urban100 public datasets show that the visual effect and Peak Signal-to-Noise Ratio （PSNR） of the proposed method are superior to those of the current mainstream algorithms.