2 个月前

图像恢复的简单基线方法

Liangyu Chen; Xiaojie Chu; Xiangyu Zhang; Jian Sun
图像恢复的简单基线方法
摘要

尽管图像恢复领域最近取得了显著进展,但最先进(SOTA)方法的系统复杂度也在不断增加,这可能会阻碍方法的便捷分析和比较。在本文中,我们提出了一种简单基线模型,该模型不仅超越了现有的最先进方法,而且计算效率更高。为了进一步简化基线模型,我们发现非线性激活函数(如Sigmoid、ReLU、GELU、Softmax等)并非必要:它们可以用乘法替代或完全移除。因此,我们从基线模型中推导出一种无非线性激活函数的网络,即NAFNet(Nonlinear Activation Free Network)。在多个具有挑战性的基准测试中,NAFNet取得了最先进水平的结果,例如在GoPro数据集上(用于图像去模糊),其PSNR值达到33.69 dB,比之前的最先进方法高出0.38 dB,而计算成本仅为前者的8.4%;在SIDD数据集上(用于图像去噪),其PSNR值达到40.30 dB,比之前的最先进方法高出0.28 dB,且计算成本不到前者的50%。代码和预训练模型已发布在https://github.com/megvii-research/NAFNet。

图像恢复的简单基线方法 | 最新论文 | HyperAI超神经