VDN
要求和依存关系
Ubuntu 16.04,CUDA 10.0
Python 3.6,Pytorch 1.1.0
详细信息(请参阅environment.yml)
图像去噪
非IID高斯噪声消除
在训练阶段,我们使用, 的源图像作为 ,并使用高斯核生成了噪声的方差图。 但是,在测试阶段,采用了三种不同的方差图来生成噪声图像,以验证VDN的一般性。
测验
python demo_test_simulation.py
训练
从以上链接下载源图像。
准备测试数据集:
python datasets/prepare_data/simulation/noise_generate_nips_niid.py
开始培训:
python train_simulation.py --simulate_dir source_imgs_path --eps2 5e-5
真实的
2022-06-26 21:54:55
125.71MB
Python
1