Dancing under the stars video denoising in starlight

Dancing under the stars: video denoising in starlight

TL;DR

A GAN-tuned physics-based 噪声建模来表示暗光下的噪声,文章表示高斯-泊松噪声在极暗条件下并不是很受用,这种条件下的噪声经常是sensor-specific的,比较难被建模。

Method

Untitled

  • 第一个步骤是训练 noise generator 和 discriminator, 它用于区分 real noise & synthetic noise. 文章用长曝 + low gain 获得noiseless 的图片, 短曝光 + 高gain 来获取noisy 的图像。 经过训练后,该noise geerator 可以获取合成的realistic noise。
  • 第二个步骤是训练denoiser,训练数据同时使用 synthetic clean 和 noisy video. 其中motion 数据是由 noise generator 生成的,noisy still是由 数据获得的。

Untitled

  • Noise Generator 包括一些比较经典的噪声建模。例如random 的 read noise 和 shot noise, FPN , 行噪声,量化噪声,周期性噪声等等。在这些噪声的基础上,再由一个UNet 来做一个调整。

Untitled

  • GAN 训练:为了使得在每一次forward中生成不同的噪声sample,选取了GAN的训练方法。文章选取了WGAN作为框架。

Untitled

  • 降噪的架构,每一个HRNET的输入是三帧图像,类似二叉树一样两两融合。

Untitled

Experiment

Untitled

  • 最后两列的比较是是否采用residue来学习,residue 增加了overfit 的风险。指标用的是KL distance。