DIP-denosing
This is a code repo for Rethinking Deep Image Prior for Denoising (ICCV 2021).
Addressing the relationship between Deep image prior and effective degrees of freedom, DIP-SURE with STE(stochestic temporal ensemble) shows reasonable result on single image denoising.
If you use any of this code, please cite the following publication:
@article{jo2021dipdenoising,
author = {Yeonsik Jo, Se young chun, and Choi, Jonghyun},
title = {Rethinking Deep Image Prior for Denoising},
booktitle = {Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV)},
month = {October},
year = {2021},
pages = {5087-5096}
}
Working environment
- TITAN Xp
- ubuntu 18.04.4
- pytorch 1.6
Note: Experimental results were not checked in other environments.
Set-up
- Make your own environment
conda create --name DIP --file requirements.txt
conda avtivate DIP
pip install tqdm
Inference
- Produce CSet9 result
bash exp_denoising.sh CSet9 <GPU ID>
- For your own data with sigma=25 setup
mkdir testset/<YOUR_DATASET>
python main.py --dip_type eSURE_new --net_type s2s --exp_tag <EXP_NAME> --optim RAdam --force_steplr --desc sigma25 denoising --sigma 25 --eval_data <YOUR_DATASET>
Browsing experimental result
- We provide reporting code with invoke.
invoke showtable csv/<exp_type>/<exp_tag>
- Example.
invoke showtable csv/poisson/MNIST/
PURE_dc_scale001_new optimal stopping : 384.30, 31.97/0.02 | ZCSC : 447.60, 31.26/0.02 | STE 31.99/0.02
PURE_dc_scale01_new optimal stopping : 94.70, 24.96/0.12 | ZCSC : 144.60, 24.04/0.14 | STE 24.89/0.12
PURE_dc_scale02_new optimal stopping : 70.30, 22.92/0.20 | ZCSC : 110.00, 21.82/0.22 | STE 22.83/0.20
<EXEPRIMENTAL NAME> optimal stopping :<STEP>, <PSNR>/<LPIPS> | ZCSC : <STEP>, <PSNR>/<LPIPS>| STE <PSNR>/<LPIPS>
The reported numbers are PSNR/LPIPS.
Results in paper
For the result used on paper, please refer this link.
SSIM score
For SSIM score of color images, I used matlab code same as the author of S2S.
This is the demo code I received from the S2S author.
Thank you Mingqin!
% examples
ref = im2double(imread('gt.png'));
noisy = im2double(imread('noisy.png'));
psnr_result = psnr(ref, noisy);
ssim_result = ssim(ref, noisy);
License
MIT license.
Contacts
For questions, please send an email to [email protected]
57 Oct 22, 2022
54 Nov 23, 2022
218 Dec 27, 2022
47 Dec 26, 2022
18 Oct 12, 2022
362 Jan 03, 2023
2.5k Jan 04, 2023
42 Oct 29, 2022
89 Jan 08, 2023
385 Dec 28, 2022
6 Jan 24, 2022
68 Nov 24, 2022
15 Feb 10, 2022
71 Jan 07, 2023
15 Jun 17, 2021
86 Dec 07, 2022
116 Jan 04, 2023
5 Sep 07, 2022
958 Jan 03, 2023
9 Oct 31, 2022