This is a PyTorch implementation of the 32nd Workshop on Image Processing and Image Understanding (IPIU 2020) paper, Residual U-shaped Network for Image Denoising.
If you find our project useful in your research, please consider citing:
@inproceedings{kim2019run,
title={Residual U-shaped Network for Image Denoising},
author={Kim, Sangmin and Park, Bumjun and Jeong, Jechang},
booktitle={32nd Workshop on Image Processing and Image Understanding (IPIU 2020)},
year={2020}
Python 3.6
PyTorch 1.4.0
We used the DIV2K datasets for training the proposed network.
Furthermore, the datasets for testing the proposed network are
BSD68 datasets and Set12 datasets that were used in DnCNN.
There are other options you can choose.
Please refer to dataset.py.
Use the following command to use our training codes
python train.py
There are other options you can choose. Please refer to train.py.
Use the following command to use our test codes
python test.py
There are other options you can choose. Please refer to test.py.
If you have any question about the code or paper, feel free to ask me to [email protected].
Thanks for SaoYan who gave the implementaion of DnCNN.
This work was supported by the BK21 PLUS(Brain Korea 21 Program for Leading Universities & Students) funded by the Ministry of Education, Korea. (H0301-14-1018)