3 个月前

低光原始去噪的可学习性增强:成对真实数据与噪声建模的融合

低光原始去噪的可学习性增强:成对真实数据与噪声建模的融合

摘要

低光照原始图像去噪是计算成像领域中一项重要且具有价值的任务,当前基于学习的方法主要依赖于成对的真实数据进行训练。然而,真实数据量有限以及噪声分布复杂,构成了成对真实数据的可学习性瓶颈,限制了基于学习方法的去噪性能。为解决这一问题,本文提出一种可学习性增强策略,通过噪声建模对成对真实数据进行重构。该策略包含两项高效技术:曝光噪声增强(Shot Noise Augmentation, SNA)与暗区阴影校正(Dark Shading Correction, DSC)。通过噪声模型解耦,SNA 通过增加数据量提升数据映射的精度,DSC 则通过降低噪声复杂度来简化数据映射过程。在公开数据集及真实成像场景下的大量实验结果表明,所提方法在去噪性能上达到了当前最优水平。代码已开源,地址为:https://github.com/megvii-research/PMN。

代码仓库

megvii-research/pmn
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
image-denoising-on-eld-sonya7s2-x100PMN
PSNR (Raw): 46.50
SSIM (Raw): 0.985
image-denoising-on-eld-sonya7s2-x200PMN
PSNR (Raw): 44.51
SSIM (Raw): 0.973
image-denoising-on-sid-sonya7s2-x100PMN
PSNR (Raw): 43.16
SSIM (Raw): 0.960
image-denoising-on-sid-sonya7s2-x250PMN
PSNR (Raw): 40.92
SSIM (Raw): 0.947
image-denoising-on-sid-x100PMN
PSNR (Raw): 43.16
SSIM: 0.960
image-denoising-on-sid-x300PMN
PSNR (Raw): 37.77
SSIM: 0.934

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
低光原始去噪的可学习性增强:成对真实数据与噪声建模的融合 | 论文 | HyperAI超神经