17 天前

用于图像修复的深度CNN去噪先验学习

用于图像修复的深度CNN去噪先验学习

摘要

基于模型的优化方法与判别式学习方法是解决低层视觉中各类逆问题的两种主流策略。通常,这两种方法各具优势与局限:基于模型的优化方法在处理不同逆问题时具有较强的灵活性,但为了获得良好性能,往往需要引入复杂的先验知识,导致计算耗时较长;而判别式学习方法虽然具有快速的推理速度,其应用范围却受限于特定任务。近期研究表明,借助变量分裂技术,可通过判别式学习获得的去噪器作为模块化组件嵌入基于模型的优化框架中,从而求解其他逆问题(如图像去模糊)。当所采用的去噪器由判别式学习方法获得时,这种融合方式能带来显著优势。然而,针对快速判别式去噪先验的融合研究仍显不足。为此,本文旨在训练一组快速且高效的卷积神经网络(CNN)去噪器,并将其集成到基于模型的优化方法中,以解决各类逆问题。实验结果表明,所学习到的去噪器集合不仅在高斯去噪任务中取得了优异性能,还可作为先验知识,有效提升多种低层视觉应用的性能。

代码仓库

cszn/ircnn
官方
pytorch

基准测试

基准方法指标
color-image-denoising-on-bsd68-sigma15Deep CNN Denoiser
PSNR: 33.86
color-image-denoising-on-bsd68-sigma25Deep CNN Denoiser
PSNR: 31.16
color-image-denoising-on-bsd68-sigma35Deep CNN Denoiser
PSNR: 29.5
color-image-denoising-on-bsd68-sigma5Deep CNN Denoiser
PSNR: 40.36
color-image-denoising-on-cbsd68-sigma50IRCNN
PSNR: 27.86
grayscale-image-denoising-on-bsd68-sigma15Deep CNN Denoiser
PSNR: 31.63
grayscale-image-denoising-on-bsd68-sigma25Deep CNN Denoiser
PSNR: 29.15
grayscale-image-denoising-on-bsd68-sigma50Deep CNN Denoiser
PSNR: 26.19
image-super-resolution-on-set14-2x-upscalingDeep CNN Denoiser
PSNR: 30.79
image-super-resolution-on-set14-3x-upscalingDeep CNN Denoiser
PSNR: 27.72
image-super-resolution-on-set14-4x-upscalingDeep CNN Denoiser
PSNR: 27.59
image-super-resolution-on-set5-2x-upscalingDeep CNN Denoiser
PSNR: 35.05
image-super-resolution-on-set5-3x-upscalingDeep CNN Denoiser
PSNR: 31.26

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
用于图像修复的深度CNN去噪先验学习 | 论文 | HyperAI超神经