Invertible Image Denoising
This is the PyTorch implementation of paper: Invertible Denoising Network: A Light Solution for Real Noise Removal (CVPR 2021). arxiv.
Dependencies and Installation
- Python 3 (Recommend to use Anaconda)
- PyTorch >= 1.0
- NVIDIA GPU + CUDA
- Python packages:
pip install numpy opencv-python lmdb pyyaml
- TensorBoard:
- PyTorch >= 1.1:
pip install tb-nightly future
- PyTorch == 1.0:
pip install tensorboardX
- PyTorch >= 1.1:
Dataset Preparation
The datasets used in this paper is DND (can be downloaded here), SIDD (can be downloaded here) and RNI.
Get Started
Training and testing codes are in 'codes/'. Please see 'codes/README.md' for basic usages. Pretrained model can be found in 'pretrained/'
Invertible Architecture
Visual Results
All visual results for SIDD dataset can be found in 'results/'.
Acknowledgement
The code is based on Invertible Image Rescaling. If you find this code is helpful, please also cite the paper Invertible Image Rescaling.
Citation
If you find this work helps you, please cite:
@article{liu2021invertible,
title={Invertible Denoising Network: A Light Solution for Real Noise Removal},
author={Liu, Yang and Qin, Zhenyue and Anwar, Saeed and Ji, Pan and Kim, Dongwoo and Caldwell, Sabrina and Gedeon, Tom},
journal={arXiv preprint arXiv:2104.10546},
year={2021}
}
Contact
If you have any questions, please contact [email protected] or [email protected].