论文标题

插件播放图像恢复与深度Denoiser先验

Plug-and-Play Image Restoration with Deep Denoiser Prior

论文作者

Zhang, Kai, Li, Yawei, Zuo, Wangmeng, Zhang, Lei, Van Gool, Luc, Timofte, Radu

论文摘要

关于插件图像修复的最新作品表明,Denoiser可以隐式作为基于模型的方法的图像,以解决许多反问题。当通过深层卷积神经网络(CNN)与大型建模能力歧视时,这种属性可引起插件图像恢复的可观优势(例如,整合基于模型的方法的灵活性和基于学习方法的有效性的灵活性)。但是,尽管更深层次的CNN模型正在迅速越来越受欢迎,但由于缺乏合适的DeNoiser先验,现有的插件图像恢复阻碍了其性能。为了突破即插即用图像修复的限制,我们通过训练高度灵活且有效的CNN Denoiser提前设置了一个基准深度Denoiser。然后,我们将Deep DeNoiser作为模块化部分插入基于半二次分裂的迭代算法,以解决各种图像恢复问题。同时,我们对参数设置,中间结果和经验收敛提供了详尽的分析,以更好地了解工作机制。在包括Deblurring,Super-Losolution和Demosaicing在内的三个代表性图像恢复任务上进行的实验结果表明,提议的插件播放图像恢复与Deep Denoiser先验不仅显着超过其他基于模型的方法,而且还可以在基于正常的学习基于的基于先进的学习方法上实现竞争力甚至卓越的性能。源代码可在https://github.com/cszn/dpir上找到。

Recent works on plug-and-play image restoration have shown that a denoiser can implicitly serve as the image prior for model-based methods to solve many inverse problems. Such a property induces considerable advantages for plug-and-play image restoration (e.g., integrating the flexibility of model-based method and effectiveness of learning-based methods) when the denoiser is discriminatively learned via deep convolutional neural network (CNN) with large modeling capacity. However, while deeper and larger CNN models are rapidly gaining popularity, existing plug-and-play image restoration hinders its performance due to the lack of suitable denoiser prior. In order to push the limits of plug-and-play image restoration, we set up a benchmark deep denoiser prior by training a highly flexible and effective CNN denoiser. We then plug the deep denoiser prior as a modular part into a half quadratic splitting based iterative algorithm to solve various image restoration problems. We, meanwhile, provide a thorough analysis of parameter setting, intermediate results and empirical convergence to better understand the working mechanism. Experimental results on three representative image restoration tasks, including deblurring, super-resolution and demosaicing, demonstrate that the proposed plug-and-play image restoration with deep denoiser prior not only significantly outperforms other state-of-the-art model-based methods but also achieves competitive or even superior performance against state-of-the-art learning-based methods. The source code is available at https://github.com/cszn/DPIR.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源