3 个月前

通过像素级噪声感知对抗训练学习生成逼真的噪声图像

通过像素级噪声感知对抗训练学习生成逼真的噪声图像

摘要

现有的深度学习真实去噪方法通常需要大量带噪声-干净图像对作为监督信号。然而,获取真实场景下的带噪声-干净图像数据集成本高昂且过程繁琐,难以实际应用。为缓解这一问题,本文研究了如何生成具有真实感的噪声图像。首先,我们提出了一种简洁而合理的噪声建模方法,将每个真实噪声像素视为一个随机变量。该模型将噪声图像生成问题分解为两个子问题:图像域对齐与噪声域对齐。随后,我们提出一种新型框架——像素级噪声感知生成对抗网络(Pixel-level Noise-aware Generative Adversarial Network, PNGAN)。PNGAN 利用预训练的真实去噪器,将生成的虚假噪声图像与真实噪声图像映射至近乎无噪的解空间,从而实现图像域对齐;同时,通过像素级对抗训练机制,完成噪声域对齐。此外,为提升噪声拟合效果,我们设计了一种高效的生成器架构——简单多尺度网络(Simple Multi-scale Network, SMNet)。定性评估表明,PNGAN 生成的噪声在强度和分布特性上与真实噪声高度相似。定量实验进一步证明,使用生成噪声图像训练的一系列去噪模型在四个真实去噪基准测试中均取得了当前最优(SOTA)性能。部分代码、预训练模型及实验结果已开源,可供对比参考,地址为:https://github.com/caiyuanhao1998/PNGAN。

基准测试

基准方法指标
image-denoising-on-namPNGAN
PSNR: 40.78
SSIM: 0.986
image-denoising-on-polyu-datasetPNGAN
PSNR: 40.55
SSIM: 0.983
image-denoising-on-siddPNGAN
PSNR (sRGB): 40.07
SSIM (sRGB): 0.960
noise-estimation-on-siddPNGAN
Average KL Divergence: 0.153
PSNR Gap: 0.84

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
通过像素级噪声感知对抗训练学习生成逼真的噪声图像 | 论文 | HyperAI超神经