国内刊号:51-1307/TP
国际刊号:1001-9081
发布日期:
作者:王素玉, 杨静, 李越
单位:1.北京市物联网软件与系统工程技术研究中心(北京工业大学),北京 100124;2.北京工业大学 信息学部,北京 100124
关键词:信息蒸馏网络,图像超分辨率复原,空间注意力,通道注意力,混合损失函数
针对超分辨率复原技术中网络层数不断加深导致的网络训练困难、特征信息利用率低等问题,设计并实现了一种基于双注意力的信息蒸馏网络(IDN)的图像超分辨率复原算法。首先,利用IDN较低的计算复杂度及信息蒸馏模块提取更多特征的优势,通过引入残差注意力模块(RAM)并考虑图像通道之间的相互依赖性来自适应地重新调整特征权重,从而进一步提升图像高分辨率细节的重建能力;然后,设计了对于边缘信息敏感的新型混合损失函数对图像进行细化处理,以加速网络收敛。在Set5、Set14、BSD100和Urban100公共数据集上的测试结果表明,该方法的视觉效果和峰值信噪比(PSNR)均优于当前主流算法。
来源:2022年第1期
《计算机应用》期刊编辑部