摘要
针对超分辨率复原技术中网络层数不断加深导致的网络训练困难、特征信息利用率低等问题,设计并实现了一种基于双注意力的信息蒸馏网络(IDN)的图像超分辨率复原算法。首先,利用IDN较低的计算复杂度及信息蒸馏模块提取更多特征的优势,通过引入残差注意力模块(RAM)并考虑图像通道之间的相互依赖性来自适应地重新调整特征权重,从而进一步提升图像高分辨率细节的重建能力;然后,设计了对于边缘信息敏感的新型混合损失函数对图像进行细化处理,以加速网络收敛。在Set5、Set14、BSD100和Urban100公共数据集上的测试结果表明,该方法的视觉效果和峰值信噪比(PSNR)均优于当前主流算法。
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.
作者
王素玉
杨静
李越
WANG Suyu;YANG Jing;LI Yue(Beijing Engineering Research Center for IoT Software and Systems(Beijing University of Technology),Beijing 100124,China;Faculty of Information Technology,Beijing University of Technology,Beijing 100124,China)
出处
《计算机应用》
CSCD
北大核心
2022年第1期239-244,共6页
journal of Computer Applications
关键词
信息蒸馏网络
图像超分辨率复原
空间注意力
通道注意力
混合损失函数
Information Distillation Network(IDN)
image super-resolution restoration
Spatial Attention(SA)
Channel Attention(CA)
mixed loss function