3 个月前

基于物理的极端低光原始图像去噪噪声生成模型

基于物理的极端低光原始图像去噪噪声生成模型

摘要

由于缺乏丰富且真实的训练数据,现有的单张图像去噪算法在处理与训练数据分布差异较大的真实原始图像时泛化能力较差。尽管通过异方差高斯噪声模型可部分缓解该问题,但数字相机电子系统所引入的噪声源仍被严重忽视,而这些噪声对原始图像测量的影响尤为显著,尤其是在极端低光照条件下。为解决这一问题,本文提出了一种基于CMOS图像传感器特性的高精度噪声生成模型,从而能够合成更符合图像形成物理过程的真实样本。在此噪声模型的基础上,我们进一步提出了一种针对现有主流数字相机的噪声参数校准方法,该方法简单、可复现,适用于任意新型设备。我们通过构建一个覆盖多个品牌现代数字相机的新低光照去噪数据集,系统性地评估了现有训练方案下神经网络的泛化性能。大量实验结果一致表明,采用本文提出的噪声生成模型后,网络的性能可达到如同在丰富真实数据上训练一般,充分验证了所提噪声模型的有效性。

代码仓库

Vandermode/NoiseModel
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
image-denoising-on-eld-sonya7s2-x100ELD
PSNR (Raw): 45.45
SSIM (Raw): 0.975
image-denoising-on-eld-sonya7s2-x200ELD
PSNR (Raw): 43.43
SSIM (Raw): 0.954

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
基于物理的极端低光原始图像去噪噪声生成模型 | 论文 | HyperAI超神经