arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.03488eess.IVcs.CV

快速、非局部且基于神经网络:一种轻量级高质量图像去噪解决方案

Fast, nonlocal and neural: a lightweight high quality solution to image denoising

School of Mathematical Science, Inner Mongolia University(内蒙古大学数学科学学院);Centre Borelli, ENS Paris-Saclay, CNRS(巴黎萨克雷高等师范学院博雷利中心、法国国家科学研究中心)

Yu Guo, Axel Davy, Gabriele Facciolo, Jean-Michel Morel, Qiyu Jin

AI总结:

针对卷积神经网络去噪计算量大、易过度平滑规则纹理的问题,提出结合非局部算法与轻量级残差CNN的去噪方案,实现低计算量下的高性能去噪,速度提升显著且在复杂纹理图像上效果突出。

摘要:

随着卷积神经网络(CNNs)的广泛应用,传统的基于模型的去噪算法性能已被超越。然而,CNNs面临两个问题:其一,它们计算需求高,这使得其在移动终端的部署尤为困难;其二,实验证据表明,与传统非局部模型相比,CNNs往往会过度平滑图像中存在的规则纹理。在这篇快报中,我们通过将非局部算法与轻量级残差CNN相结合,提出了一种同时解决这两个问题的方案。该方案可充分发挥两种模型的优势。我们将此框架应用于经典非局部算法(NLM和BM3D)的两种GPU实现中,观察到两种情况下性能均有大幅提升,在低计算需求下表现优于现有最优方法。我们的方案比性能相当的CNNs快10到20倍,并且能获得更高的峰值信噪比(PSNR)。此外,最终方法在包含复杂纹理的图像(如MIT Moire数据集的图像)上表现出显著提升。

英文摘要:

With the widespread application of convolutional neural networks (CNNs), the traditional model based denoising algorithms are now outperformed. However, CNNs face two problems. First, they are computationally demanding, which makes their deployment especially difficult for mobile terminals. Second, experimental evidence shows that CNNs often over-smooth regular textures present in images, in contrast to traditional non-local models. In this letter, we propose a solution to both issues by combining a nonlocal algorithm with a lightweight residual CNN. This solution gives full latitude to the advantages of both models. We apply this framework to two GPU implementations of classic nonlocal algorithms (NLM and BM3D) and observe a substantial gain in both cases, performing better than the state-of-the-art with low computational requirements. Our solution is between 10 and 20 times faster than CNNs with equivalent performance and attains higher PSNR. In addition the final method shows a notable gain on images containing complex textures like the ones of the MIT Moire dataset.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑