3 个月前

MSSNet:用于单张图像去模糊的多尺度阶段网络

MSSNet:用于单张图像去模糊的多尺度阶段网络

摘要

在深度学习出现之前,大多数传统的单图像去模糊方法采用“粗到细”(coarse-to-fine)的策略,即先在粗分辨率下估计出一个清晰图像,再逐步在更精细的尺度上进行优化。尽管这一策略也被应用于部分基于深度学习的方法中,但近年来,一些单尺度方法相继提出,其在图像质量与计算效率方面均展现出优于传统“粗到细”方法的性能。本文重新审视“粗到细”框架,深入分析了以往该类方法中存在的缺陷,这些缺陷严重制约了其性能表现。基于上述分析,我们提出了一种新型的基于深度学习的单图像去模糊方法——多尺度阶段网络(Multi-Scale-Stage Network, MSSNet),通过引入针对性的改进措施以克服原有方法的不足。具体而言,MSSNet创新性地引入了三个关键技术组件:反映模糊尺度的阶段配置机制、跨尺度信息传播机制,以及基于像素洗牌(pixel-shuffle)的多尺度处理方案。大量实验结果表明,MSSNet在图像质量、模型规模和计算效率三个方面均达到了当前最优水平,实现了性能的全面领先。

代码仓库

kky7/MSSNet
官方
pytorch

基准测试

基准方法指标
deblurring-on-goproMSSNet-small
PSNR: 32.02
SSIM: 0.953
deblurring-on-goproMSSNet
PSNR: 33.01
SSIM: 0.961
deblurring-on-goproMSSNet-large
PSNR: 33.39
SSIM: 0.964
deblurring-on-realblur-j-1MSSNet
PSNR (sRGB): 32.1
Params(M): 15.6
SSIM (sRGB): 0.928
deblurring-on-realblur-j-trained-on-goproMSSNet
PSNR (sRGB): 28.79
SSIM (sRGB): 0.879
deblurring-on-realblur-rMSSNet
PSNR (sRGB): 39.76
Params: 15.59
SSIM (sRGB): 0.972
deblurring-on-realblur-r-trained-on-goproMSSNet
PSNR (sRGB): 35.93
SSIM (sRGB): 0.953

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
MSSNet:用于单张图像去模糊的多尺度阶段网络 | 论文 | HyperAI超神经