
摘要
单图像超分辨率(SISR)随着卷积神经网络(CNN)的深度和宽度不断增加,已经取得了显著进展。然而,庞大的参数量阻碍了其在实际问题中的应用。在本文中,我们提出了一种轻量级特征融合网络(LFFN),该网络能够在充分利用多尺度上下文信息的同时,大幅减少网络参数,从而最大化SISR的效果。LFFN基于纺锤块和一个自适应学习的Softmax特征融合模块(SFFM)。具体而言,纺锤块由维度扩展单元、特征探索单元和特征精炼单元组成。维度扩展层将低维特征扩展到高维,并隐式地学习适合下一单元的特征图。特征探索单元对不同的特征图进行线性和非线性的特征探索。特征精炼层用于融合和精炼特征。SFFM通过Softmax函数以自适应学习的方式融合来自不同模块的特征,从而在少量参数成本的情况下充分利用层次信息。基准数据集上的定性和定量实验表明,LFFN在参数量相似的情况下,性能优于现有最先进的方法。
代码仓库
qibao77/LFFN-master
官方
tf
GitHub 中提及
qibao77/LFFN
tf
GitHub 中提及
基准测试
| 基准 | 方法 | 指标 |
|---|---|---|
| image-super-resolution-on-bsd100-2x-upscaling | LFFN-S | PSNR: 31.96 |
| image-super-resolution-on-bsd100-3x-upscaling | LFFN-S | PSNR: 28.91 |
| image-super-resolution-on-bsd100-4x-upscaling | LFFN-S | PSNR: 27.42 |
| image-super-resolution-on-manga109-2x | LFFN-S | PSNR: 37.93 SSIM: 0.9746 |
| image-super-resolution-on-manga109-3x | LFFN-S | PSNR: 32.8 SSIM: 0.9381 |
| image-super-resolution-on-manga109-4x | LFFN-S | PSNR: 29.76 SSIM: 0.8979 |
| image-super-resolution-on-set5-2x-upscaling | LFFN-S | PSNR: 37.66 SSIM: 0.9585 |
| image-super-resolution-on-set5-3x-upscaling | LFFN-S | PSNR: 34.04 SSIM: 0.9233 |