3 个月前

基于可学习离散小波变换的高效多尺度网络用于盲运动去模糊

基于可学习离散小波变换的高效多尺度网络用于盲运动去模糊

摘要

在传统的单图像运动去模糊方法中,粗到精(coarse-to-fine)策略被广泛采用;然而,在深度学习背景下,现有的多尺度算法不仅需要借助复杂的模块来融合低分辨率RGB图像与深层语义特征,还需人工生成低分辨率图像对,而这些图像对往往缺乏足够的可信度。针对上述问题,本文提出一种基于单输入多输出(Single-Input Multiple-Output, SIMO)架构的多尺度网络,用于运动去模糊任务,有效简化了传统粗到精框架的算法复杂性。为缓解多尺度结构带来的细节信息恢复缺陷,我们结合真实世界模糊轨迹的特性,引入可学习的小波变换模块,重点捕捉模糊图像到清晰图像之间逐步过渡过程中的方向连续性与频率特征。最终,本文提出一种基于可学习离散小波变换(Learnable Discrete Wavelet Transform, MLWNet)的多尺度网络,在多个真实世界去模糊数据集上均取得了当前最优的性能表现,无论在主观视觉质量、客观评价指标,还是计算效率方面均展现出显著优势。

代码仓库

thqiu0419/mlwnet
官方
pytorch

基准测试

基准方法指标
deblurring-on-goproMLWNet
PSNR: 33.83
SSIM: 0.968
deblurring-on-realblur-j-1MLWNet
PSNR (sRGB): 33.84
SSIM (sRGB): 0.941
deblurring-on-realblur-rMLWNet
PSNR (sRGB): 40.69
SSIM (sRGB): 0.976
deblurring-on-rsblurMLWNet
Average PSNR: 34.94
SSIM: 0.880

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于可学习离散小波变换的高效多尺度网络用于盲运动去模糊 | 论文 | HyperAI超神经