3 个月前

亚像素反投影网络用于轻量级单图像超分辨率

亚像素反投影网络用于轻量级单图像超分辨率

摘要

基于卷积神经网络(CNN)的方法在单图像超分辨率(SISR)任务中取得了显著成功。然而,大多数现有模型在提升重建精度的同时,往往导致模型参数数量大幅增加,进而提高了计算资源的需求。为应对这一挑战,本文致力于在保持超分辨率重建性能的前提下,减少基于CNN的SISR方法的参数量与计算开销。为此,我们提出了一种新型网络架构,能够在重建质量与低计算复杂度之间实现良好平衡。具体而言,我们设计了一种基于子像素卷积(sub-pixel convolution)替代反卷积(deconvolution)层的迭代反投影架构。通过在四个主流超分辨率基准数据集上进行广泛的定量与定性评估,我们验证了所提模型在计算效率与重建精度方面的优越性能。实验结果表明,与当前最先进的SISR方法相比,本文提出的方法在显著减少模型参数量和计算成本的同时,仍能保持优异的重建精度。代码已公开,获取地址为:https://github.com/supratikbanerjee/SubPixel-BackProjection_SuperResolution。

代码仓库

基准测试

基准方法指标
image-super-resolution-on-bsds100-2xSPBP-L+
PSNR: 32.21
SSIM: 0.9001
image-super-resolution-on-set14-2x-upscalingSPBP-L+
PSNR: 33.62
SSIM: 0.9178
image-super-resolution-on-set5-2x-upscalingSPBP-L+
PSNR: 38.05
SSIM: 0.9606
image-super-resolution-on-urban100-2xSPBP-L+
PSNR: 32.07
SSIM: 0.9277

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
亚像素反投影网络用于轻量级单图像超分辨率 | 论文 | HyperAI超神经