Skip to content

Pytorch implementation of Learning to See in the Dark in CVPR 2018

Notifications You must be signed in to change notification settings

huyvnphan/Learning_To_See_In_The_Dark

Repository files navigation

Learning to See in the Dark

This is my Pytorch implementation of Learning to See in the Dark in CVPR 2018. For more information, please read the Report.pdf and See-In-The-Dark notebook.

Project Website

Original paper by Chen Chen, Qifeng Chen, Jia Xu, and Vladlen Koltun.

Original Tensorflow implementation by Chen Chen

Results

PSNR on test set No. Epochs
Chen et al. 28.88 4000
My implementation 28.83 4000

Training PSNR and L1 Loss Training PSNR and L1 Loss

Denoise a random image in original dataset Denoise a random image in original dataset

Denoise a Sony A7sii raw image found on Google Denoise a random image in original dataset

Testing cross sensor generalizaion

This image was taken on a Sony RX100VI with a 1-inch Bayer sensor Cross Sensor

Requirements

  • 64GB RAM
  • Pytorch 1.0
  • Numpy + Rawpy
  • Matplotlib

I ran the code on a Tesla V100 32GB. If your GPU has less memory, reduce the batch size.

Releases

No releases published

Packages

No packages published